Correct parameters in the Traffic Scripts Reformer.sh and Transformer.sh

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

Correct parameters in the Traffic Scripts Reformer.sh and Transformer.sh #111

Closed dg1609 closed 1 year ago

dg1609 commented 1 year ago

To fix the following error when trying to reproduce training for the provided traffic dataset:

RuntimeError: Given groups=1, weight of size [512, 1, 3], expected input[32, 862, 98] to have 1 channels, but got 862 channels instead