Method create

gsplat/compression_simulation/mask.py:238–247

(config: MaskConfig, device: Optional[torch.device])

class AdaptiveMaskFactory:
    @staticmethod
    def create(config: MaskConfig, device: Optional[torch.device]) -> AdaptiveMaskBase:
        if not config.enabled or config.strategy is None:
            return NullAdaptiveMask()
        if config.strategy == "learnable":
            if device is None:
                device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
            return LearnableAdaptiveMask(config, device=device)
        if config.strategy == "gradient":
            return GradientAdaptiveMask(config)
        raise ValueError(f"Unsupported mask strategy '{config.strategy}'")