Update exp_main.py

HyeongD · Nov 21, 2022 · e1bfb66 · e1bfb66
1 parent e116bbc
commit e1bfb66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/exp/exp_main.py b/exp/exp_main.py
@@ -144,7 +144,7 @@ def train(self, setting):
                     if self.args.output_attention:
                         outputs = self.model(batch_x, batch_x_mark, dec_inp, batch_y_mark)[0]
                     else:
-                        outputs = self.model(batch_x, batch_x_mark, dec_inp, batch_y_mark, batch_y)
+                        outputs = self.model(batch_x, batch_x_mark, dec_inp, batch_y_mark)
 
                     f_dim = -1 if self.args.features == 'MS' else 0
                     outputs = outputs[:, -self.args.pred_len:, f_dim:]
@@ -315,4 +315,4 @@ def predict(self, setting, load=False):
 
         np.save(folder_path + 'real_prediction.npy', preds)
 
-        return
+        return