Hello everyone.
How can I convert a model thats been trained on multiple GPUs (wrapped in DataParallel) to cpu?
I tried  model = model.module.to('cpu') without any luck.
How should I be going about this?
Notes:
pn1:  I’m trying to save this as a jit traced model thats why I’m trying t o convert it to cpu before hand
pn2:  I know I can simply use model.module in torch.jit.trace and then map the weights to cpu when loading  the jit model. however, I want to know why this is failing here!