agolynski:

in


Thanks for your response. I was able to solve the issue by passing the loaded model through DataParallel again.
model_single = MyModel()
#solution
model = nn.DataParallel(model_single)
model = model.to(device)
#end
checkpoint = torch.load(checkpoint_path, map_location="cpu")
model.load_state_dict(checkpoint) #ERROR HERE