There was an error while loading. Please reload this page.
1 parent 046abef commit 50afe29Copy full SHA for 50afe29
time_sequence_prediction/train.py
@@ -39,7 +39,7 @@ def forward(self, input, future = 0):
39
np.random.seed(0)
40
torch.manual_seed(0)
41
# load data and make training set
42
- data = torch.load(open('traindata.pt'))
+ data = torch.load('traindata.pt')
43
input = Variable(torch.from_numpy(data[3:, :-1]), requires_grad=False)
44
target = Variable(torch.from_numpy(data[3:, 1:]), requires_grad=False)
45
# build the model
0 commit comments