IMvision12 commited on
Commit
c5e5dc2
·
verified ·
1 Parent(s): 931076e

Upgrade kf_config.json to v2 (nested config + weight_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.nextvit",
5
  "model_class": "NextViTImageClassify",
6
  "variant": "nextvit_base_bd_in1k",
7
  "weights": "model.weights.h5",
 
 
8
  "model_type": "nextvit",
9
- "depths": [
10
- 3,
11
- 4,
12
- 20,
13
- 3
14
- ],
15
- "stem_chs": [
16
- 64,
17
- 32,
18
- 64
19
- ],
20
- "head_dim": 32,
21
- "mix_block_ratio": 0.75,
22
- "sr_ratios": [
23
- 8,
24
- 4,
25
- 2,
26
- 1
27
- ],
28
- "drop_path_rate": 0.1,
29
- "image_size": 224,
30
- "num_classes": 1000
 
 
31
  }
 
1
  {
2
  "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.nextvit",
5
  "model_class": "NextViTImageClassify",
6
  "variant": "nextvit_base_bd_in1k",
7
  "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "weight_dtype": "float32",
10
  "model_type": "nextvit",
11
+ "vision_config": {
12
+ "depths": [
13
+ 3,
14
+ 4,
15
+ 20,
16
+ 3
17
+ ],
18
+ "stem_chs": [
19
+ 64,
20
+ 32,
21
+ 64
22
+ ],
23
+ "head_dim": 32,
24
+ "mix_block_ratio": 0.75,
25
+ "sr_ratios": [
26
+ 8,
27
+ 4,
28
+ 2,
29
+ 1
30
+ ],
31
+ "drop_path_rate": 0.1,
32
+ "image_size": 224,
33
+ "num_classes": 1000
34
+ }
35
  }