open without 'rb' caused Python 3 to open this in text mode and fail (p…

…ytorch#133)
b-schnell · Apr 7, 2017 · 50afe29 · 50afe29
1 parent 046abef
commit 50afe29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/time_sequence_prediction/train.py b/time_sequence_prediction/train.py
@@ -39,7 +39,7 @@ def forward(self, input, future = 0):
     np.random.seed(0)
     torch.manual_seed(0)
     # load data and make training set
-    data = torch.load(open('traindata.pt'))
+    data = torch.load('traindata.pt')
     input = Variable(torch.from_numpy(data[3:, :-1]), requires_grad=False)
     target = Variable(torch.from_numpy(data[3:, 1:]), requires_grad=False)
     # build the model