MappingConfig

data class MappingConfig(val usePathBasedMatching: Boolean = true, val fallbackToShapeMatching: Boolean = true, val debug: Boolean = false, val nameResolver: WeightNameResolver? = null)(source)

Configuration for weight mapping behavior.

Constructors

Link copied to clipboard
constructor(usePathBasedMatching: Boolean = true, fallbackToShapeMatching: Boolean = true, debug: Boolean = false, nameResolver: WeightNameResolver? = null)

Properties

Link copied to clipboard

Whether to print debug information during mapping

Link copied to clipboard

Whether to fall back to shape-only matching if name matching fails

Link copied to clipboard

Optional resolver that translates module paths/param names to tensor names. When provided, this takes priority over the default ONNX-style path matching.

Link copied to clipboard

Whether to use DSL module path for name matching