Informer-PyTorch-Lightning

This is a reorganized implementation of Informer based on the official implementation and ⚡ Lightning.

Requirements

• numpy
• pandas
• scikit-learn
• torch
• lightning>=2.0
• torchmetrics>=0.11

⚠️ The repository is currently based on Lightning 2.0. To use PyTorch Lightning v1.x, please switch to the pl_v1 branch.

Model Training

# template
python main.py --config configs/{DATASET_NAME}/{multi/uni}variate/pred_len_{PRED_LEN}.yaml

# for example
python main.py --config configs/ETTh1/multivariate/pred_len_24.yaml