Update tiny models for EfficientFormerForImageClassification

config.json

@@ -1,16 +1,18 @@
 {
+  "_name_or_path": "tiny_models/efficientformer/EfficientFormerForImageClassification",
   "architectures": [
     "EfficientFormerForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "attention_ratio": 4,
+  "batch_norm_eps": 1e-05,
   "depths": [
-    3,
     2,
-    6,
-    4
+    2,
+    2,
+    2
   ],
-  "dim": 448,
+  "dim": 128,
   "distillation": true,
   "downsample_pad": 1,
   "downsample_patch_size": 3,
@@ -25,29 +27,29 @@
   "encoder_stride": 2,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
-  "hidden_size": 448,
+  "hidden_size": 128,
   "hidden_sizes": [
-    48,
-    96,
-    224,
-    448
+    16,
+    32,
+    64,
+    128
   ],
-  "image_size": 224,
+  "image_size": 64,
   "initializer_range": 0.02,
   "intermediate_size": 37,
   "key_dim": 32,
   "layer_norm_eps": 1e-12,
   "layer_scale_init_value": 1e-05,
-  "mlp_expansion_ratio": 4,
+  "mlp_expansion_ratio": 2,
   "model_type": "efficientformer",
-  "num_attention_heads": 8,
+  "num_attention_heads": 4,
   "num_channels": 3,
   "num_hidden_layers": 7,
   "num_meta3d_blocks": 1,
   "patch_size": 2,
   "pool_size": 3,
-  "resolution": 7,
+  "resolution": 2,
   "torch_dtype": "float32",
-  "transformers_version": "4.28.0.dev0",
+  "transformers_version": "4.31.0.dev0",
   "use_layer_scale": true
 }

preprocessor_config.json

@@ -1,7 +1,7 @@
 {
   "crop_size": {
-    "height": 224,
-    "width": 224
+    "height": 64,
+    "width": 64
   },
   "do_center_crop": true,
   "do_normalize": true,
@@ -21,7 +21,7 @@
   "resample": 3,
   "rescale_factor": 0.00392156862745098,
   "size": {
-    "height": 224,
-    "width": 224
+    "height": 64,
+    "width": 64
   }
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:52ee27f06678a90465d0e1beb6f1da50a99346b0130769ab92c0b545b97f3910
-size 45819689
+oid sha256:2124748eb8901447d7737dc00b45ee69d3300db5dd20f786437b1a1c25253926
+size 1843305

tf_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e1d6737290525d6b0af57318f03256217680b330a6848261178bbb534ff495d
+size 1951984