hi,i am loading the model under DataParallel,below is my code:
#training 
model = build_model()
model = nn.DataParallel(model)
model.cuda()
....training...
torch.save(model,name)

#eval
net = torch.load(name)

i save the whole model under DataParallel form,not just state_dict,and i load the model directly for eval,i am wondering is this right?will it get the same performance as save the state_dict and load the state_dict?  thank you so much!!