Change worker_num to 4 to speed up tipc training (#7328)

314019c0 · WangZhen · GitHub · 111ae8b5 · 314019c0
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt .../yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt +1 -1

未找到文件。
--- a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt
+++ b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt
@@ -13,7 +13,7 @@ train_infer_img_dir:./dataset/coco/test2017/
 filename:null
 ##
 trainer:norm_train
-norm_train:tools/train.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml -o
+norm_train:tools/train.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml -o worker_num=4
 pact_train:tools/train.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml --slim_config configs/slim/quant/yolov3_darknet_qat.yml -o
 fpgm_train:tools/train.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml --slim_config configs/slim/prune/yolov3_darknet_prune_fpgm.yml -o
 distill_train:null