Bug
跑时TransGTR-main/train_forecast.py
时,出现RuntimeError: Expected 2D (unbatched) or 3D (batched) input to conv1d
大家说
法一:将ConV-1d都换成了ConV-2d
法二:配置老版本的环境 pytorch=1.10(TransGTR-main
指定的torch版本是1.9.0,但是我装不上……)
Debug
选择法二
我的cuda版本是11.6
在torch历史版本这里查看,发现可以安装torch==1.10.0+cu113
用下面的命令安装成功
pip install torch==1.10.0+cu113 torchvision==0.11.0+cu113 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html