loadfunc

models/cmp/utils/common_utils.py

@@ -96,7 +96,8 @@ def load_state(path, model, optimizer=None):
         return storage.cuda()
     if os.path.isfile(path):
         print("=> loading checkpoint '{}'".format(path))
-        checkpoint = torch.load(path, map_location=map_func)
+        # checkpoint = torch.load(path, map_location=map_func)
+        checkpoint = torch.load(path)
         model.load_state_dict(checkpoint['state_dict'], strict=False)
         ckpt_keys = set(checkpoint['state_dict'].keys())
         own_keys = set(model.state_dict().keys())