Update config.json

config.json

@@ -3,20 +3,19 @@
     "EfficientNetForImageClassification"
   ],
   "depths": [
-    3,
-    4,
-    6,
-    3
-  ],
-  "downsample_in_first_stage": false,
-  "embedding_size": 64,
-  "hidden_act": "relu",
-  "hidden_sizes": [
-    256,
-    512,
-    1024,
-    2048
+    1,
+    1,
+    1,
+    1
   ],
+  "num_classes": 1000,
+  "width_coefficient": 1.0,
+  "depth_coefficient": 1.0,
+  "dropout_rate": 0.2,
+  "image_size": 224,
+  "mean": [0.485, 0.456, 0.406],
+  "std": [0.229, 0.224, 0.225]
+}
   "id2label": {
     "0": "tench, Tinca tinca",
     "1": "goldfish, Carassius auratus",
@@ -2023,6 +2022,6 @@
   "layer_type": "bottleneck",
   "model_type": "efficientnet",
   "num_channels": 3,
-  "torch_dtype": "float32",
+  "dtype": "float32",
   "transformers_version": "4.18.0.dev0"
 }