my old model trained in just two gpu and now i get two more gpu,so I want to change the device_ids of my model but met some problem.
I have save the model so I want to use
to load my model but it doesn’t work.
The error come to be ‘all tensors must be on devices’.
So I try to use nn.DataParallel after load the model,but it lead to a DataParallel in DataParallel which seem to be strange,and I got RuntimeError: all tensors must be on devices again.
Does only one can help me?I just want to change my model’s device_ids .