e-bug / volta

[TACL 2021] Code and data for the framework in "Multimodal Pretraining Unmasked: A Meta-Analysis and a Unified Framework of Vision-and-Language BERTs"
https://aclanthology.org/2021.tacl-1.58/
MIT License
114 stars 24 forks source link

multiple GPUS #17

Closed nooralahzadeh closed 2 years ago

nooralahzadeh commented 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.

Thanks

e-bug commented 2 years ago

You can try passing --drop_last to so that each GPU always processes the same batch size.

nooralahzadeh commented 2 years ago

Thanks