KerasHub
Divyasreepat commited on
Commit
89d5b3e
1 Parent(s): c20383a

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. README.md +18 -0
  2. config.json +52 -0
  3. metadata.json +6 -0
  4. model.weights.h5 +3 -0
README.md ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: keras-hub
3
+ ---
4
+ This is a [`MiT` model](https://keras.io/api/keras_hub/models/mi_t) uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
5
+ Model config:
6
+ * **name:** mi_t_backbone
7
+ * **trainable:** True
8
+ * **depths:** [2, 2, 2, 2]
9
+ * **hidden_dims:** [64, 128, 320, 512]
10
+ * **image_shape:** [224, 224, 3]
11
+ * **num_layers:** 4
12
+ * **blockwise_num_heads:** [1, 2, 5, 8]
13
+ * **blockwise_sr_ratios:** [8, 4, 2, 1]
14
+ * **max_drop_path_rate:** 0.1
15
+ * **patch_sizes:** [7, 3, 3, 3]
16
+ * **strides:** [4, 2, 2, 2]
17
+
18
+ This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.
config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.mix_transformer.mix_transformer_backbone",
3
+ "class_name": "MiTBackbone",
4
+ "config": {
5
+ "name": "mi_t_backbone",
6
+ "trainable": true,
7
+ "depths": [
8
+ 2,
9
+ 2,
10
+ 2,
11
+ 2
12
+ ],
13
+ "hidden_dims": [
14
+ 64,
15
+ 128,
16
+ 320,
17
+ 512
18
+ ],
19
+ "image_shape": [
20
+ 224,
21
+ 224,
22
+ 3
23
+ ],
24
+ "num_layers": 4,
25
+ "blockwise_num_heads": [
26
+ 1,
27
+ 2,
28
+ 5,
29
+ 8
30
+ ],
31
+ "blockwise_sr_ratios": [
32
+ 8,
33
+ 4,
34
+ 2,
35
+ 1
36
+ ],
37
+ "max_drop_path_rate": 0.1,
38
+ "patch_sizes": [
39
+ 7,
40
+ 3,
41
+ 3,
42
+ 3
43
+ ],
44
+ "strides": [
45
+ 4,
46
+ 2,
47
+ 2,
48
+ 2
49
+ ]
50
+ },
51
+ "registered_name": "keras_hub>MiTBackbone"
52
+ }
metadata.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "keras_version": "3.6.0",
3
+ "keras_hub_version": "0.16.1",
4
+ "parameter_count": 13151424,
5
+ "date_saved": "2024-10-08@18:17:32"
6
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15435a2f62afde81b2ccfc4b188a04ed00c2f688c57198b04762a80d8eebbdb6
3
+ size 52975904