webnn
/

efficientnet-lite4

Image Classification

Inference Endpoints

Model card Files Files and versions Community

captainspock commited on May 8, 2024

Commit

f979de7

·

verified ·

1 Parent(s): 0c021d5

Update config.json

Files changed (1) hide show

config.json +4 -4

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
-  "_name_or_path": "microsoft/resnet-50",
   "architectures": [
-    "ResNetForImageClassification"
   ],
   "depths": [
     3,
@@ -2022,7 +2022,7 @@
     "zucchini, courgette": 939
   },
   "layer_type": "bottleneck",
-  "model_type": "resnet",
   "num_channels": 3,
   "out_features": [
     "stage4"
@@ -2038,4 +2038,4 @@
     "stage4"
   ],
   "transformers_version": "4.34.0.dev0"
-}

 {
+  "_name_or_path": "google/efficientnet-b7",
   "architectures": [
+    "EfficientNetForImageClassification"
   ],
   "depths": [
     3,
     "zucchini, courgette": 939
   },
   "layer_type": "bottleneck",
+  "model_type": "efficientnet",
   "num_channels": 3,
   "out_features": [
     "stage4"
     "stage4"
   ],
   "transformers_version": "4.34.0.dev0"
+}