[Fix] fix the num_workers setting (#25)

2022-09-18 15:38:36 +08:00 · 2022-09-18 15:38:36 +08:00 · d3fed9f2cd
parent 33de37a252
commit d3fed9f2cd
2 changed files with 2 additions and 0 deletions
--- a/configs/yolov5/yolov5_s-v61_syncbn_fast_1xb4-300e_balloon.py
+++ b/configs/yolov5/yolov5_s-v61_syncbn_fast_1xb4-300e_balloon.py
@ -14,6 +14,7 @@ metainfo = {

 train_dataloader = dict(
    batch_size=train_batch_size_per_gpu,
+    num_workers=train_num_workers,
    dataset=dict(
        data_root=data_root,
        metainfo=metainfo,
--- a/docs/zh_cn/user_guides/yolov5_tutorial.md
+++ b/docs/zh_cn/user_guides/yolov5_tutorial.md
@ -55,6 +55,7 @@ metainfo = {

 train_dataloader = dict(
    batch_size=train_batch_size_per_gpu,
+    num_workers=train_num_workers,
    dataset=dict(
        data_root=data_root,
        metainfo=metainfo,