import gin

@gin.configurable
class StripNetModelsConfig:
    def __init__(
        self,
        stripnet_path: str = "nn_models/STRIPNET/stripnet_s.pth",
        stripnet_freeze: bool = True,
        stripnet_mona_last_n_stages: int = 2,
        stripnet_backbone_lr_factor: float = 0.1,
    ) -> None:
        self.stripnet_path = stripnet_path
        self.stripnet_freeze = stripnet_freeze
        self.stripnet_mona_last_n_stages = stripnet_mona_last_n_stages
        self.stripnet_backbone_lr_factor = stripnet_backbone_lr_factor
        
