Update Infer.py

Jeevesh8 · May 12, 2020 · c55316c · c55316c
1 parent aed40b3
commit c55316c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Infer.py b/Infer.py
@@ -181,8 +181,7 @@ def predict_next(t, date_lis, test_dataset) :
         from Models import LSTM
         t = LSTM.lstm(seq_len = args.seq_len, ini_len=args.ini_len, final_len=model_final_len).to(device)
 
-    if path.exists(args.param_file) :
-        t.load_state_dict(torch.load(args.param_file))
+    t.load_state_dict(torch.load(args.param_file))
 
     t = t.double()