RePST 修复版 准备GPT-2预训练权重 ```bash mkdir GPT-2 wget https://huggingface.co/openai-community/gpt2/resolve/main/config.json?download=true -O ./GPT-2/config.json wget https://huggingface.co/openai-community/gpt2/resolve/main/pytorch_model.bin?download=true -O ./GPT-2/pytorch_model.bin ```` 准备PEMS-BAY数据集,按照[BasicTS](https://github.com/GestaltCogTeam/BasicTS/blob/master/tutorial/dataset_design.md)方法准备 [Google Drive](https://drive.google.com/drive/folders/14EJVODCU48fGK0FkyeVom_9lETh80Yjp?usp=sharing) 可使用gdown下载。 解压后,确保 `./datasets/PEMS-BAY` 文件夹内具有 `adj_mx.pkl, data.dat, desc,json文件`, 然后运行脚本 ```bash python prepare_pems_bay.py ```` 在PEMS-BAY数据集文件夹下生成 `train.npz, val.npz, test.npz` 根据BasicTS仓库配置BasicTS环境,亦或是使用 `pip install -r requirements.txt` 我是直接使用现有的BasicTS环境,因此没有做过测试 创建log文件夹到项目仓库 ```bash mkdir log ``` 开跑 ```python python run.py --root_path datasets --data_path PEMS-BAY --device cuda:0 --seq_len 24 --pred_len 24 ```