Changed model_type in config.json and MusheffConfig class to musheff

README.md

@@ -58,7 +58,7 @@ Here is how to use this model to classify a mushroom image into one of the 12 Ru
 ```python
 import torch
 from datasets import load_dataset
-from transformers import AutoImageProcessor
+from transformers import AutoModel, AutoImageProcessor
 
 
 dataset = load_dataset("SoFa325/12_popular_russia_mushrooms_edible_poisonous")

config.json

@@ -2,7 +2,10 @@
   "_name_or_path": "blasisd/musheff",
   "architectures": ["Musheff"],
   "dropout_rate": 0.3,
-  "model_type": "efficientnet_b3",
+  "model_type": "musheff",
+  "backbone_config": {
+    "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 = "efficientnet_b3"
+    model_type = "musheff"
 
     def __init__(self, num_classes=12, dropout_rate=0.3, **kwargs):
         self.num_classes = num_classes