andrea-gatto
Results
2
comments of
andrea-gatto
@hlky why did you close this?
It looks like that the problem resides in a number of `.cuda()` and `device="cuda"` lines throughout the code. This sends the model/models to the first GPU while sometimes `to.device(correct_gpu)` send...