Closed nooralahzadeh closed 2 years ago
Hi, Did you run the code on multiple GPUS, I am getting the following error Caught StopIteration in replica 0 on device 0.
Caught StopIteration in replica 0 on device 0.
Thanks
You can try passing --drop_last to so that each GPU always processes the same batch size.
--drop_last
Hi, Did you run the code on multiple GPUS, I am getting the following error
Caught StopIteration in replica 0 on device 0.
Thanks