search

lucidrains / iTransformer

Unofficial implementation of iTransformer - SOTA Time Series Forecasting using Attention networks, out of Tsinghua / Ant group
MIT License
445 stars 36 forks source link

How to train iTransformer? #27

Open MrDr-Pan opened 4 months ago

MrDr-Pan commented 4 months ago

Thanks for sharing the codes, and the paper is great! I've noticed that this project is only an architecture of iTransformer, how can I train the model?

satyrmipt commented 2 months ago

@lucidrains, please answer, do your package uses pretrained models from original git (article author's git) or training loop implemented inside model call? Is preds = model(time_series) just apply trained model or train it and then apply to time_series?