Hi! So I do this:
net = torch.nn.DataParallel(AutoEncoder(config), device_ids = list_of_gpus)
But then when I print:  for elem in net.parameters(): print(elem.device)
I get a list of “cpu” for most of hte parameters, and at the end I get cuda:0. Why is this the case? Thanks