In the file dataset/data_loader.py, you have set num_workers=batch_size in the function make_ddp_train_loader. Is there any specific reason for this? You mentioned in issue #13 that you've done this for convenience. Can you please elaborate on this?
This looks like it should instead be num_workers=train.cfg.num_workers. Please let me know if this is correct.
In the file dataset/data_loader.py, you have set
num_workers=batch_size
in the functionmake_ddp_train_loader
. Is there any specific reason for this? You mentioned in issue #13 that you've done this for convenience. Can you please elaborate on this?This looks like it should instead be
num_workers=train.cfg.num_workers
. Please let me know if this is correct.Thank you!