update x_mark embedding

salmanali88 · Apr 17, 2024 · c6cdf82 · c6cdf82
1 parent 032706f
commit c6cdf82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/exp/exp_long_term_forecasting.py b/exp/exp_long_term_forecasting.py
@@ -47,7 +47,7 @@ def vali(self, vali_data, vali_loader, criterion):
         with torch.no_grad():
             for i, (batch_x, batch_y, batch_x_mark, batch_y_mark) in enumerate(vali_loader):
                 batch_x = batch_x.float().to(self.device)
-                batch_y = batch_y.float()
+                batch_y = batch_y.float().to(self.device)
 
                 batch_x_mark = batch_x_mark.float().to(self.device)
                 batch_y_mark = batch_y_mark.float().to(self.device)