Class QuantizerConfig

gsplat/compression_simulation/config.py:87–118

Configuration for differentiable quantization.

@dataclass
class QuantizerConfig:
    """Configuration for differentiable quantization."""

    attributes: Dict[str, AttributeQuantizerConfig] = field(
        default_factory=_default_attribute_configs
    )

    def copy_for_attribute_overrides(
        self, overrides: Optional[Mapping[str, Mapping[str, Any]]]
    ) -> "QuantizerConfig":
        if not overrides:
            return self
        updated = {
            name: AttributeQuantizerConfig(
                enabled=config.enabled,
                bitwidth=config.bitwidth,
                clamp_range=config.clamp_range,
                warmup_steps=config.warmup_steps,
                warmup_bitwidth=config.warmup_bitwidth,
                mode=config.mode,
            )
            for name, config in self.attributes.items()
        }
        for name, cfg in overrides.items():
            if name not in updated:
                continue
            attr_cfg = updated[name]
            for key, value in cfg.items():
                if not hasattr(attr_cfg, key):
                    raise KeyError(f"Unknown quantizer setting '{name}.{key}'")
                setattr(attr_cfg, key, value)
        return QuantizerConfig(attributes=updated)


@dataclass