fix error again

styleTransfer.py

@@ -66,27 +66,27 @@ vgg.eval()
 
 new_state_dict = OrderedDict()
 state_dict = torch.load(decoder_path)
-# for k, v in state_dict.items():
-#     #namekey = k[7:] # remove `module.`
-#     namekey = k
-#     new_state_dict[namekey] = v
-decoder.load_state_dict(state_dict)
+for k, v in state_dict.items():
+    #namekey = k[7:] # remove `module.`
+    namekey = k
+    new_state_dict[namekey] = v
+decoder.load_state_dict(new_state_dict)
 
 new_state_dict = OrderedDict()
 state_dict = torch.load(Trans_path)
-# for k, v in state_dict.items():
-#     #namekey = k[7:] # remove `module.`
-#     namekey = k
-#     new_state_dict[namekey] = v
-Trans.load_state_dict(state_dict)
+for k, v in state_dict.items():
+    #namekey = k[7:] # remove `module.`
+    namekey = k
+    new_state_dict[namekey] = v
+Trans.load_state_dict(new_state_dict)
 
 new_state_dict = OrderedDict()
 state_dict = torch.load(embedding_path)
-# for k, v in state_dict.items():
-#     #namekey = k[7:] # remove `module.`
-#     namekey = k
-#     new_state_dict[namekey] = v
-embedding.load_state_dict(state_dict)
+for k, v in state_dict.items():
+    #namekey = k[7:] # remove `module.`
+    namekey = k
+    new_state_dict[namekey] = v
+embedding.load_state_dict(new_state_dict)
 
 network = StyTR.StyTrans(vgg,decoder,embedding,Trans)
 network.eval()