-
Notifications
You must be signed in to change notification settings - Fork 8
/
config.example.ini
150 lines (138 loc) · 4.67 KB
/
config.example.ini
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
[DEFAULT]
# [Optional] MRT Model Root Directory, `~/mrt_model` by default.
Model_dir=
# [Required]
# Automatically download the model from gluon-cv into Model_dir,
# please refer to gluon-cv model zoo for more available models,
# or train and save model into Model_dir manually.
Model_name=
# [Optional] string values: cpu(default) or gpu.
Device_type=
# [Optional] Int string split by comma, such as 3,4,5 or 3,
# given Device_type equals to gpu.
Device_ids=
# [Optional] Control the logger hiearchy of print debug information.
# int values: 0(None) by default, 10(DEBUG),
# 20(INFO), 30(WARNING), 40(ERROR), 50(CRITICAL).
Verbosity=
# [Required] Input tuple shape array
# -1 stands for the axis of batch size, (-1, 3, 224, 224) etc.
Input_shape=
# [Optional] Mrt starting checkpoint.
# it's useful for developer to debug with the `Dump` key
# in section belows and restart last quantization procedure
# without repeat calibration etc.
#
# string values: DEFAULT by default, PREPARE,
# SPLIT_MODEL, CALIBRATION, QUANTIZATION, MERGE_MODEL.
#
# 0. prerequisites for `DEFAULT` checkpoint
# [model_name].json
# [model_name].params
# 1. prerequisites for `PREPARE` checkpoint
# [model_name].prepare.json
# [model_name].prepare.params
# 2. prerequisites for `SPLIT_MODEL` checkpoint (with `Split_keys`)
# [model_name].base.json
# [model_name].base.params
# [model_name].top.json
# [model_name].top.params
# 3. prerequisites for `CALIBRATION` checkpoint
# [model_name].mrt.calibrate.json
# [model_name].mrt.calibrate.params
# [model_name].mrt.calibrate.ext
# [model_name].top.json (with `Split_keys`)
# [model_name].top.params (with `Split_keys`)
# 4. prerequisites for `QUANTIZATION` checkpoint
# [model_name].mrt.quantize.json
# [model_name].mrt.quantize.params
# [model_name].mrt.quantize.ext
# [model_name].top.json (with `Split_keys`)
# [model_name].top.params (with `Split_keys`)
# 5. prerequisites for `MERGE_MODEL` checkpoint (with `Split_keys`)
# [model_name].all.quantize.json
# [model_name].all.quantize.params
# [model_name].all.quantize.ext
Start=
[PREPARE]
# [Optional] Mrt checkpoint.
# bool values: True or False(default).
Dump=
# Split model into two subgraph, usually for detection model.
[SPLIT_MODEL]
# [Optional] Model internal names of spliting keys.
Keys=
# [Optional] Mrt checkpoint.
# bool values: True or False(default).
Dump=
[CALIBRATION]
# [Required] Available dataset list:
# imagenet, voc, trec, mnist, quickdraw and cifar10.
Dataset=
# [Optional] Calibration batch size, 16 by default.
Batch=
# [Optional] Iterator numbers of calibration, 1 by default.
Calibrate_num=
# [Optional] Hyperparameter for the threshold of model internal data,
# by default it's absolute max of data, or mean+lambda*std.
Lambda=
# [Optional] Dataset root directory for specific dataset out of
# list above.
Dataset_dir=
# [Optional] Derived from the context in section `DEFAULT`.
Device_type=
Device_ids=
# [Optional] Mrt checkpoint.
# bool values: True or False(default).
Dump=
[QUANTIZATION]
# [Optional] The precision of the mrt input precision.
Input_precision=
# [Optional] The precision of the mrt output precision.
Output_precision=
# [Optional] Derived from the context in section `CALIBRATION`.
Device_type=
Device_ids=
# [Optional] Hyperparameter for the quantization of Softmax operator,
# by default it's 10.
Softmax_lambd=
# [Optional] Hyperparameter for the quantization of requant function,
# by default it's 5.
Shift_bits=
# [Optional] Inital threshold for base symbol given model split.
# FORMAT [node name]: [threshold value
Thresholds=
# [Optional] Debug usage
Restore_name=
# [Optional] Mrt checkpoint.
# bool values: True or False(default).
Dump=
[MERGE_MODEL]
# [Optional] Adjust the top attributes with
# respect to the base oscales.
#
# FORMAT [op_name]: [the name of attribute]: [original base name]
Attribute_deps=
# [Optional] Model merger output scales name map.
# FORMAT [quantized base name]: [original base name]
Oscale_maps=
# [Optional] Mrt checkpoint.
# bool values: True or False(default).
Dump=
# [Optional] section, can be commentted.
[EVALUATION]
# [Optional] Derived from last batch size by default.
# the `Batch` must be divisible by the number of gpus in `Device_ids`.
Batch=
# [Optional] Derived from the context in section `QUANTIZATION` by default.
Device_type=
Device_ids=
# [Optional] Iterator numbers of dataset evaluation, 0 by default.
Iter_num=
# [Optional] section, can be commentted.
[COMPILATION]
# Derived from last batch size by default.
Batch=
# The compilation dump directory, derived from section
# `DEFAULT` Model_dir by default.
Dump_dir=