Class MaskConfig

gsplat/compression_simulation/config.py:154–182

Configuration for adaptive mask stage.

@dataclass
class MaskConfig:
    """Configuration for adaptive mask stage."""

    enabled: bool = False
    strategy: Optional[str] = "learnable"
    start_step: int = 10_000
    regularization_weight: float = 1.0
    cap_max: Optional[int] = None
    learnable: LearnableMaskSettings = field(default_factory=LearnableMaskSettings)
    gradient: GradientMaskSettings = field(default_factory=GradientMaskSettings)

    def to_dict(self) -> Dict[str, Any]:
        return {
            "enabled": self.enabled,
            "strategy": self.strategy,
            "start_step": self.start_step,
            "regularization_weight": self.regularization_weight,
            "cap_max": self.cap_max,
            "learnable": {
                "start_temp": self.learnable.start_temp,
                "end_temp": self.learnable.end_temp,
                "total_iters": self.learnable.total_iters,
                "target_sparsity": self.learnable.target_sparsity,
                "lr": self.learnable.lr,
            },
            "gradient": {
                "grad_threshold": self.gradient.grad_threshold,
            },
        }


@dataclass