Hi,
I have another issue.
So I have placed model.to(device) before calling nn.DataParallel. However, get this error:
RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cuda:1


I’m assuming everything needs to be on cuda0 before splitting to the other devices. How do I fix that?
Cheers,
Taran