IMvision12 commited on
Commit
6a1be70
·
verified ·
1 Parent(s): f1cef21

Upgrade kf_config.json to v2 (nested config + dtype)

Browse files
Files changed (1) hide show
  1. kf_config.json +27 -23
kf_config.json CHANGED
@@ -1,31 +1,35 @@
1
  {
2
  "library_name": "kerasformers",
3
- "kerasformers_version": "1.1.3",
4
  "model_module": "kerasformers.models.maxvit",
5
  "model_class": "MaxViTImageClassify",
6
  "variant": "maxvit_tiny_tf_384_in1k",
7
  "weights": "model.weights.h5",
 
 
8
  "model_type": "maxvit",
9
- "stem_width": 64,
10
- "depths": [
11
- 2,
12
- 2,
13
- 5,
14
- 2
15
- ],
16
- "embed_dim": [
17
- 64,
18
- 128,
19
- 256,
20
- 512
21
- ],
22
- "num_heads": [
23
- 2,
24
- 4,
25
- 8,
26
- 16
27
- ],
28
- "window_size": 12,
29
- "image_size": 384,
30
- "num_classes": 1000
 
 
31
  }
 
1
  {
2
  "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.maxvit",
5
  "model_class": "MaxViTImageClassify",
6
  "variant": "maxvit_tiny_tf_384_in1k",
7
  "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "dtype": "float32",
10
  "model_type": "maxvit",
11
+ "vision_config": {
12
+ "stem_width": 64,
13
+ "depths": [
14
+ 2,
15
+ 2,
16
+ 5,
17
+ 2
18
+ ],
19
+ "embed_dim": [
20
+ 64,
21
+ 128,
22
+ 256,
23
+ 512
24
+ ],
25
+ "num_heads": [
26
+ 2,
27
+ 4,
28
+ 8,
29
+ 16
30
+ ],
31
+ "window_size": 12,
32
+ "image_size": 384,
33
+ "num_classes": 1000
34
+ }
35
  }