KerasHub
Divyasreepat commited on
Commit
9bcb07c
1 Parent(s): 5d0ec0a

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:** [3, 6, 40, 3]
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
+ 3,
9
+ 6,
10
+ 40,
11
+ 3
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": 81443008,
5
+ "date_saved": "2024-10-08@18:19:03"
6
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b337714ec8624376299d6c3619428374d28c58286daf571925203ebdbdf523a
3
+ size 328032744