RuntimeError: module must have its parameters and buffers on device cuda:1 (device_ids[0]) but found one of them on device: cpu

Question

I am implementing nn.DataParallel class in code and i encapsulate all my models in this class. It gives an error ambiguous error.

Class Code