IMvision12 commited on
Commit
37141d4
·
verified ·
1 Parent(s): 1c197c9

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

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