search

thuml / Autoformer

About Code release for "Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting" (NeurIPS 2021), https://arxiv.org/abs/2106.13008
MIT License
2k stars 429 forks source link

exp_main:test/predict 中把label也作为decoder的输入了 #177

Closed deng113jie closed 1 year ago

deng113jie commented 1 year ago

https://github.com/thuml/Autoformer/blob/ab192380693bd04a1dc8aa6ff85addefcf0cf909/exp/exp_main.py#L206

是不是少了target mask？

wuhaixu2016 commented 1 year ago

您好，请见：https://github.com/thuml/Autoformer/blob/ab192380693bd04a1dc8aa6ff85addefcf0cf909/exp/exp_main.py#L212C88-L212C88 会在后面步骤中将batch_y遮住，所以不会有未来泄漏的问题