Skip to content

Latest commit

 

History

History
26 lines (18 loc) · 1.46 KB

README.md

File metadata and controls

26 lines (18 loc) · 1.46 KB

Informer-PyTorch-Lightning

GitHub stars issues License Maintenance Code style: black Codefactor

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