| from transformers import PretrainedConfig | |
| class EfficientNetV25Config(PretrainedConfig): | |
| model_type = "efficientnetv25" | |
| def __init__(self, | |
| url="https://huggingface.co/FredZhang7/efficientnetv2.5_rw_s/resolve/main/efficientnetv2.5_base_in1k.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) |