Update models/transformer/encode_decode/model.py

models/transformer/encode_decode/model.py

@@ -63,7 +63,7 @@ class EncoderDecoder(nn.Module):
     @classmethod
     def load_from_file(cls, file_path):
         # Load model
-        checkpoint = torch.load(file_path, map_location='cuda:0')
+        checkpoint = torch.load(file_path, map_location='cpu')
         para_dict = checkpoint['model_parameters']
         vocab_size = para_dict['vocab_size']
         model = EncoderDecoder.make_model(vocab_size, vocab_size, para_dict['N'],