IMvision12 commited on
Commit
faebaf8
·
verified ·
1 Parent(s): 57dc00a

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

Browse files
Files changed (1) hide show
  1. kf_config.json +16 -4
kf_config.json CHANGED
@@ -1,12 +1,24 @@
1
  {
2
  "library_name": "kerasformers",
3
- "kerasformers_version": "1.1.3",
4
  "model_module": "kerasformers.models.mobilevitv2",
5
  "model_class": "MobileViTV2ImageClassify",
6
  "variant": "mobilevitv2_175_cvnets_in1k",
7
  "weights": "model.weights.h5",
 
 
8
  "model_type": "mobilevitv2",
9
- "multiplier": 1.75,
10
- "image_size": 256,
11
- "num_classes": 1000
 
 
 
 
 
 
 
 
 
 
12
  }
 
1
  {
2
  "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.mobilevitv2",
5
  "model_class": "MobileViTV2ImageClassify",
6
  "variant": "mobilevitv2_175_cvnets_in1k",
7
  "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "weight_dtype": "float32",
10
  "model_type": "mobilevitv2",
11
+ "vision_config": {
12
+ "multiplier": 1.75,
13
+ "image_size": 256,
14
+ "num_classes": 1000,
15
+ "output_stride": 32,
16
+ "atrous_rates": [
17
+ 6,
18
+ 12,
19
+ 18
20
+ ],
21
+ "aspp_out_channels": 512,
22
+ "aspp_dropout_prob": 0.1
23
+ }
24
  }