IMvision12 commited on
Commit
75593c8
·
verified ·
1 Parent(s): 5c80263

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

Browse files
Files changed (1) hide show
  1. kf_config.json +19 -15
kf_config.json CHANGED
@@ -1,23 +1,27 @@
1
  {
2
  "library_name": "kerasformers",
3
- "kerasformers_version": "1.1.3",
4
  "model_module": "kerasformers.models.mit",
5
  "model_class": "MiTImageClassify",
6
  "variant": "mit_b1_in1k",
7
  "weights": "model.weights.h5",
 
 
8
  "model_type": "mit",
9
- "embed_dim": [
10
- 64,
11
- 128,
12
- 320,
13
- 512
14
- ],
15
- "depths": [
16
- 2,
17
- 2,
18
- 2,
19
- 2
20
- ],
21
- "image_size": 224,
22
- "num_classes": 1000
 
 
23
  }
 
1
  {
2
  "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.mit",
5
  "model_class": "MiTImageClassify",
6
  "variant": "mit_b1_in1k",
7
  "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "weight_dtype": "float32",
10
  "model_type": "mit",
11
+ "vision_config": {
12
+ "embed_dim": [
13
+ 64,
14
+ 128,
15
+ 320,
16
+ 512
17
+ ],
18
+ "depths": [
19
+ 2,
20
+ 2,
21
+ 2,
22
+ 2
23
+ ],
24
+ "image_size": 224,
25
+ "num_classes": 1000
26
+ }
27
  }