maazie
/

EfficientNetB0

Image Classification

Model card Files Files and versions

maazie commited on May 20, 2023

Commit

eee53dd

·

1 Parent(s): 6ab64e5

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "ResNetForImageClassification"
   ],
   "depths": [
     3,
@@ -2021,7 +2021,7 @@
     "zucchini, courgette": 939
   },
   "layer_type": "bottleneck",
-  "model_type": "resnet",
   "num_channels": 3,
   "torch_dtype": "float32",
   "transformers_version": "4.18.0.dev0"

 {
   "architectures": [
+    "EfficientNetForImageClassification"
   ],
   "depths": [
     3,
     "zucchini, courgette": 939
   },
   "layer_type": "bottleneck",
+  "model_type": "efficientnet",
   "num_channels": 3,
   "torch_dtype": "float32",
   "transformers_version": "4.18.0.dev0"