Update config.json

config.json

@@ -1,6 +1,7 @@
 {
   "architecture":"resnext101_32x8d",
-  "num_labels": 5,
-  "image_size": 416,
-  "pooler_type": "avg"
+  "num_classes": 5,
+  "pretrained_cfg":{
+    "input_size":[3,416,416],
+  }
 }