Upload config

config.json

@@ -1,6 +1,7 @@
-{
-  "architectures": ["ResNet50"],
-
-  "num_labels": 6,
-  "input_size": [3, 224, 224]
-}
+{
+  "auto_map": {
+    "AutoConfig": "config.Resnet50Config"
+  },
+  "num_classes": 6,
+  "transformers_version": "4.37.2"
+}

config.py

@@ -0,0 +1,14 @@
+from transformers import PretrainedConfig
+
+class Resnet50Config(PretrainedConfig):
+    # since we have an image classification task
+    # we need to put a model type that is close to our task
+    # don't worry this will not affect our model
+    #model_type = "MobileNetV1"
+    def __init__(
+        self,
+        num_classes=6,
+        **kwargs):
+
+        self.num_classes = num_classes
+        super().__init__(**kwargs)