forked from openvinotoolkit/nncf
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ssd512_vgg_voc_int8.json
43 lines (43 loc) · 998 Bytes
/
ssd512_vgg_voc_int8.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
{
"model": "ssd_vgg",
"batchnorm": true,
"input_info": {
"sample_size": [8, 3, 512, 512]
},
"num_classes": 21,
"dataset": "voc",
"preprocessing": {
"mean": [0.406, 0.456, 0.485],
"std": [0.255, 0.224, 0.229],
"normalize_coef": 255,
"rgb": true
},
"epochs": 78,
"batch_size": 64,
"save_freq": 3,
"optimizer": {
"type": "Adam",
"base_lr": 1e-4,
"weight_decay": 5e-4,
"schedule_type": "multistep",
"steps": [2]
},
"ssd_params": {
"steps": [8, 16, 32, 64, 128, 256, 512],
"min_sizes": [35.84, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8],
"max_sizes": [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.6],
"aspect_ratios": [[2], [2, 3], [2, 3], [2, 3], [2, 3], [2], [2]],
"variance": [0.1, 0.1, 0.2, 0.2],
"clip": false,
"flip": true
},
"compression": {
"algorithm": "quantization",
"initializer": {
"range": {
"num_init_samples": 640
}
}
},
"export_to_ir_via_onnx": true
}