forked from openvinotoolkit/nncf
-
Notifications
You must be signed in to change notification settings - Fork 1
/
ssd300_vgg_voc_int8.json
46 lines (46 loc) · 1.02 KB
/
ssd300_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
44
45
46
{
"model": "ssd_vgg",
"batchnorm": true,
"input_info": {
"sample_size": [8, 3, 300, 300]
},
"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": 24,
"batch_size": 128,
"test_interval": 1,
"save_freq": 3,
"optimizer": {
"type": "Adam",
"base_lr": 1e-5,
"weight_decay": 5e-4,
"schedule_type": "multistep",
"steps": [2]
},
"ssd_params": {
"clip": false,
"variance": [0.1, 0.1, 0.2, 0.2],
"max_sizes": [60, 111, 162, 213, 264, 315],
"min_sizes": [30, 60, 111, 162, 213, 264],
"steps": [8, 16, 32, 64, 100, 300],
"aspect_ratios": [[2], [2, 3], [2, 3], [2, 3], [2], [2]],
"flip": true,
"top_k": 200,
"loss_inference": false
},
"compression": {
"algorithm": "quantization",
"initializer": {
"range": {
"num_init_samples": 1280
}
}
},
"export_to_ir_via_onnx": true
}