How to train a model on 2 GPUs
zhanghousen opened this issue · comments
I'm trying to train a model on two GPUs but I get the following error.
IndexError: index 16 is out of bounds for dimension 0 with size 16.
How can I solve this problem?
Looking forward to your answer!