from transformers import PretrainedConfig

class MobileNetV1Config(PretrainedConfig):
    model_type = "mobilenet_custom"

    def __init__(
        self,
        num_classes: int = 200,
        block_dropout: float = 0.0,
        final_dropout: float = 0.50,
        **kwargs
    ):
        self.num_classes = num_classes
        self.block_dropout = block_dropout
        self.final_dropout = final_dropout
            
        super().__init__(**kwargs)