17 lines
498 B
YAML
17 lines
498 B
YAML
use_gpu: true
|
|
use_xpu: false
|
|
use_mlu: false
|
|
use_npu: false
|
|
log_iter: 20
|
|
save_dir: output
|
|
snapshot_epoch: 1
|
|
print_flops: false
|
|
print_params: false
|
|
|
|
# Exporting the model
|
|
export:
|
|
post_process: True # Whether post-processing is included in the network when export model.
|
|
nms: True # Whether NMS is included in the network when export model.
|
|
benchmark: False # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
|
|
fuse_conv_bn: False
|