Update config.json

config.json

@@ -1,11 +1,5 @@
 {
-  "_name_or_path": "test",
-  "depths": [
-    3,
-    4,
-    6,
-    3
-  ],
+  
     "num_labels": 5,
   "id2label": {
     "0": "Brick",
@@ -21,34 +15,5 @@
     "Plastic": 3,
     "Wood": 4
   },
-  "downsample_in_bottleneck": false,
-  "downsample_in_first_stage": false,
-  "embedding_size": 64,
-  "freeze": true,
-  "hidden_act": "relu",
-  "hidden_sizes": [
-    256,
-    512,
-    1024,
-    2048
-  ],
-  "layer_type": "bottleneck",
-  "model_type": "resnet",
-  "num_channels": 3,
-  "out_features": [
-    "stage4"
-  ],
-  "out_indices": [
-    4
-  ],
-  "outputs": 5,
-  "pretrained": true,
-  "stage_names": [
-    "stem",
-    "stage1",
-    "stage2",
-    "stage3",
-    "stage4"
-  ],
   "transformers_version": "4.42.4"
 }