Missing key(s) in state_dict

[cuongnc220592]

Hi author @Adamdad

I have a problem when testing with ResUnet.

device = torch.device("cuda" if torch.cuda.is_available() else "cpu") model = ResUNet().to(device) #if torch.cuda.device_count() > 1: model = nn.DataParallel(model).to(device) model_path = './checkpoint/ResUnet/best.pth.tar' print("=> loading checkpoint '{}'".format(model_path)) checkpoint = torch.load(model_path, map_location="cpu")

The output here. Could you please guide me how to fix these problem. Thank you very much, Best.