Added fixes in config, preprocessor and modeling

config.json

@@ -1,11 +1,12 @@
 {
   "_name_or_path": "blasisd/musheff",
   "architectures": ["Musheff"],
-  "dropout_rate": 0.3,
-  "model_type": "musheff",
-  "backbone_config": {
-    "model_type": "efficientnet_b3"
+  "auto_map": {
+    "AutoConfig": "configuration_musheff.MusheffConfig",
+    "AutoModelForImageClassification": "modeling.Musheff"
   },
+  "dropout_rate": 0.3,
+  "model_type": "efficientnet_b3",
   "num_classes": 12,
   "label2id": {
     "Agaricus_bisporus_edible": "0",

configuration_musheff.py

@@ -3,7 +3,7 @@ from transformers import PretrainedConfig
 
 
 class MusheffConfig(PretrainedConfig):
-    model_type = "musheff"
+    model_type = "efficientnet_b3"
 
     def __init__(self, num_classes=12, dropout_rate=0.3, **kwargs):
         self.num_classes = num_classes

modeling.py

@@ -30,7 +30,3 @@ class Musheff(PreTrainedModel):
 
     def forward(self, x):
         return self.model(x)
-
-
-AutoConfig.register("musheff", MusheffConfig)
-AutoModel.register(MusheffConfig, Musheff)

preprocessor_config.json

@@ -1,16 +1,29 @@
 {
+  "crop_size": {
+    "height": 289,
+    "width": 289
+  },
+  "do_center_crop": false,
+  "do_normalize": true,
+  "do_rescale": true,
   "do_resize": true,
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_processor_type": "EfficientNetImageProcessor",
+  "image_std": [
+    0.47853944,
+    0.4732864,
+    0.47434163
+  ],
+  "include_top": true,
+  "resample": 0,
+  "rescale_factor": 0.00392156862745098,
+  "rescale_offset": false,
   "size": {
-    "height": 320,
-    "width": 320
-  },
-  "resample": 3,
-  "do_center_crop": true,
-  "crop_size": {
     "height": 300,
     "width": 300
-  },
-  "do_normalize": true,
-  "image_mean": [0.485, 0.456, 0.406],
-  "image_std": [0.229, 0.224, 0.225]
+  }
 }