Solved by Ramchandra_Cheke in post #2 
              
              
                # create new OrderedDict that does not contain `module.`
from collections import OrderedDict

# original saved file with DataParallel
ckpt = torch.load("pretraied_model.pth")
state_dict = ckpt['model']    # incase there are extra parameters in the model

new_state_dict = OrderedDict()

for k, v in s…