from transformers import PretrainedConfig

class EfficientNetV25Config(PretrainedConfig):
    model_type = "efficientnetv25"

    def __init__(self,
                url="https://huggingface.co/FredZhang7/efficientnetv25_rw_s/resolve/main/pytorch_model.pth",
                input_size=[3, 304, 304],
                num_classes=1000,
                interpolation="bicubic",
                mean=[0.5, 0.5, 0.5],
                std=[0.5, 0.5, 0.5],
                **kwargs):
        self.url = url
        self.input_size = input_size
        self.num_classes = num_classes
        self.interpolation = interpolation
        self.mean = mean
        self.std = std
        super().__init__(**kwargs)