-
Notifications
You must be signed in to change notification settings - Fork 445
/
Copy pathconfiguration.yaml
281 lines (281 loc) · 7.09 KB
/
configuration.yaml
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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
description: Configuration for an instance segmentation task
header: Configuration for an instance segmentation task
learning_parameters:
batch_size:
affects_outcome_of: TRAINING
default_value: 5
description:
The number of training samples seen in each iteration of training.
Increasing this value improves training time and may make the training more
stable. A larger batch size has higher memory requirements.
editable: true
header: Batch size
max_value: 512
min_value: 1
type: INTEGER
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: 5
visible_in_ui: true
warning:
Increasing this value may cause the system to use more memory than available,
potentially causing out of memory errors, please update with caution.
auto_hpo_state: NOT_POSSIBLE
description: Learning Parameters
header: Learning Parameters
learning_rate:
affects_outcome_of: TRAINING
default_value: 0.01
description:
Increasing this value will speed up training convergence but might
make it unstable.
editable: true
header: Learning rate
max_value: 0.1
min_value: 1.0e-07
type: FLOAT
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: 0.01
visible_in_ui: true
warning: null
auto_hpo_state: NOT_POSSIBLE
learning_rate_warmup_iters:
affects_outcome_of: TRAINING
default_value: 100
description: ""
editable: true
header: Number of iterations for learning rate warmup
max_value: 10000
min_value: 0
type: INTEGER
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: 100
visible_in_ui: true
warning: null
num_checkpoints:
affects_outcome_of: NONE
default_value: 5
description: ""
editable: true
header: Number of checkpoints that is done during the single training round
max_value: 100
min_value: 1
type: INTEGER
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: 5
visible_in_ui: true
warning: null
num_iters:
affects_outcome_of: TRAINING
default_value: 1
description:
Increasing this value causes the results to be more robust but training
time will be longer.
editable: true
header: Number of training iterations
max_value: 100000
min_value: 1
type: INTEGER
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: 1
visible_in_ui: true
warning: null
num_workers:
affects_outcome_of: NONE
default_value: 5
description:
Increasing this value might improve training speed however it might
cause out of memory errors. If the number of workers is set to zero, data loading
will happen in the main training thread.
editable: true
header: Number of cpu threads to use during batch generation
max_value: 8
min_value: 0
type: INTEGER
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: 0
visible_in_ui: true
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
postprocessing:
confidence_threshold:
affects_outcome_of: INFERENCE
default_value: 0.35
description:
This threshold only takes effect if the threshold is not set based
on the result.
editable: true
header: Confidence threshold
max_value: 1
min_value: 0
type: FLOAT
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
# value: 0.35
value: 0.01
visible_in_ui: true
warning: null
description: Postprocessing
header: Postprocessing
result_based_confidence_threshold:
affects_outcome_of: INFERENCE
default_value: true
description: Confidence threshold is derived from the results
editable: true
header: Result based confidence threshold
type: BOOLEAN
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: true
visible_in_ui: true
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
algo_backend:
description: parameters for algo backend
header: Algo backend parameters
train_type:
affects_outcome_of: NONE
default_value: Incremental
description: Quantization preset that defines quantization scheme
editable: false
enum_name: TrainType
header: train type
options:
Incremental: "Incremental"
type: SELECTABLE
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: Incremental
visible_in_ui: True
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
type: CONFIGURABLE_PARAMETERS
visible_in_ui: true
pot_parameters:
description: POT Parameters
header: POT Parameters
preset:
affects_outcome_of: NONE
default_value: Performance
description: Quantization preset that defines quantization scheme
editable: True
enum_name: POTQuantizationPreset
header: Preset
options:
MIXED: Mixed
PERFORMANCE: Performance
type: SELECTABLE
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: Performance
visible_in_ui: True
warning: null
stat_subset_size:
affects_outcome_of: NONE
default_value: 300
description: Number of data samples used for post-training optimization
editable: True
header: Number of data samples
max_value: 9223372036854775807
min_value: 1
type: INTEGER
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: 300
visible_in_ui: True
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
nncf_optimization:
description: Optimization by NNCF
header: Optimization by NNCF
enable_quantization:
affects_outcome_of: INFERENCE
default_value: True
description: Enable quantization algorithm
editable: false
header: Enable quantization algorithm
type: BOOLEAN
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: true
visible_in_ui: false
warning: null
enable_pruning:
affects_outcome_of: INFERENCE
default_value: false
description: Enable filter pruning algorithm
editable: true
header: Enable filter pruning algorithm
type: BOOLEAN
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: false
visible_in_ui: true
warning: null
maximal_accuracy_degradation:
affects_outcome_of: NONE
default_value: 1.0
description: The maximal allowed accuracy metric drop in absolute values
editable: True
header: Maximum accuracy degradation
max_value: 100.0
min_value: 0.0
type: FLOAT
ui_rules:
action: DISABLE_EDITING
operator: AND
rules: []
type: UI_RULES
value: 1.0
visible_in_ui: True
warning: null
type: PARAMETER_GROUP
visible_in_ui: True