mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] fix the num_workers setting (#25)

pull/26/head
古月闻星 2022-09-18 15:38:36 +08:00 committed by GitHub
parent 33de37a252
commit d3fed9f2cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
configs/yolov5/yolov5_s-v61_syncbn_fast_1xb4-300e_balloon.py
View File

@ -14,6 +14,7 @@ metainfo = {
train_dataloader = dict( train_dataloader = dict(
batch_size=train_batch_size_per_gpu, batch_size=train_batch_size_per_gpu,
num_workers=train_num_workers,
dataset=dict( dataset=dict(
data_root=data_root, data_root=data_root,
metainfo=metainfo, metainfo=metainfo,

1
docs/zh_cn/user_guides/yolov5_tutorial.md
View File

@ -55,6 +55,7 @@ metainfo = {
train_dataloader = dict( train_dataloader = dict(
batch_size=train_batch_size_per_gpu, batch_size=train_batch_size_per_gpu,
num_workers=train_num_workers,
dataset=dict( dataset=dict(
data_root=data_root, data_root=data_root,
metainfo=metainfo, metainfo=metainfo,