pytorch>> Informer-PyTorch-Lightning>> 返回
项目作者： martinwhl

项目描述 ：
A reorganized implementation of Informer based on PyTorch Lightning
高级语言： Python
项目地址: git://github.com/martinwhl/Informer-PyTorch-Lightning.git
创建时间： 2021-04-08T07:01:30Z
项目社区：https://github.com/martinwhl/Informer-PyTorch-Lightning
开源协议：Apache License 2.0
下载


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

  1. # template
  2. python main.py --config configs/{DATASET_NAME}/{multi/uni}variate/pred_len_{PRED_LEN}.yaml
  4. # for example
  5. python main.py --config configs/ETTh1/multivariate/pred_len_24.yaml