diff --git a/configs/yolov5/yolov5_s-v61_syncbn_fast_1xb4-300e_balloon.py b/configs/yolov5/yolov5_s-v61_syncbn_fast_1xb4-300e_balloon.py index 52ebf99c..c1c57b28 100644 --- 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, diff --git a/docs/zh_cn/user_guides/yolov5_tutorial.md b/docs/zh_cn/user_guides/yolov5_tutorial.md index 15a2c9e6..2ca714a7 100644 --- 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,