kd13
/

Modern-MobileNet

Image Classification

mobilenet_custom

Model card Files Files and versions

kd13 commited on 29 days ago

Commit

73d1ecf

·

verified ·

1 Parent(s): 18d5c02

Update config.json

Files changed (1) hide show

config.json +5 -5

config.json CHANGED Viewed

@@ -1,13 +1,13 @@
 {
   "architectures": [
-    "SqueezeNetForImageClassification"
   ],
   "auto_map": {
-    "AutoConfig": "configuration_squeezenet.SqueezeNetConfig",
-    "AutoModelForImageClassification": "modeling_squeezenet.SqueezeNetForImageClassification"
   },
   "final_dropout": 0.5,
-  "fire_dropout": 0.0,
   "id2label": {
     "0": "goldfish",
     "1": "European fire salamander",
@@ -412,7 +412,7 @@
     "slug": 198,
     "orange": 199
   },
-  "model_type": "squeezenet_custom",
   "num_classes": 200,
   "torch_dtype": "float32"
 }

 {
   "architectures": [
+    "MobileNetForImageClassification"
   ],
   "auto_map": {
+    "AutoConfig": "configuration_mobilenet.MobileNetConfig",
+    "AutoModelForImageClassification": "modeling_mobilenet.MobileNetForImageClassification"
   },
   "final_dropout": 0.5,
+  "block_dropout": 0.0,
   "id2label": {
     "0": "goldfish",
     "1": "European fire salamander",
     "slug": 198,
     "orange": 199
   },
+  "model_type": "mobilenet_custom",
   "num_classes": 200,
   "torch_dtype": "float32"
 }