Class AttributeQuantizerConfig

gsplat/compression_simulation/config.py:51–69

Quantization controls for a single attribute.

@dataclass
class AttributeQuantizerConfig:
    """Quantization controls for a single attribute."""

    enabled: bool = True
    bitwidth: Optional[int] = None
    clamp_range: Optional[Tuple[float, float]] = None
    warmup_steps: Optional[int] = 10_000
    warmup_bitwidth: Optional[int] = 8
    mode: str = "noise"

    def to_dict(self) -> Dict[str, Any]:
        return {
            "enabled": self.enabled,
            "bitwidth": self.bitwidth,
            "clamp_range": list(self.clamp_range) if self.clamp_range is not None else None,
            "warmup_steps": self.warmup_steps,
            "warmup_bitwidth": self.warmup_bitwidth,
            "mode": self.mode,
        }


def _default_attribute_configs() -> Dict[str, AttributeQuantizerConfig]: