代码疑问 exp_long_term_forecasting.py

thuml / iTransformer

Official implementation for "iTransformer: Inverted Transformers Are Effective for Time Series Forecasting" (ICLR 2024 Spotlight), https://openreview.net/forum?id=JePfAI8fah

https://arxiv.org/abs/2310.06625

MIT License

1.17k stars 206 forks source link

代码疑问 exp_long_term_forecasting.py #84

Closed Sunlandis closed 3 months ago

Sunlandis commented 3 months ago

您好，exp_long_term_forecasting.py中45行 batch_y在验证的时候没有放入设备，但是下面batch_y_mark却放入设备了，猜测是否是45行最后漏写了.to(self.device)？还是有意为之呢？有什么设计作用呢？

WenWeiTHU commented 3 months ago

感谢您的建议，此处是因为推理是在cpu上进行的，所以不需要移入到gpu等设备上，并不影响实验结果。

为保证代码一致，我们已经修复了这个issue：https://github.com/thuml/iTransformer/commit/257d5c39e1a8baa27904b4215a5459674249d4ab