-
Notifications
You must be signed in to change notification settings - Fork 56
Description
Hello, I have replicated the conda environment following environment.md and downloaded the pre-trained model checkpoints, when I tried to run run_finetune_xxxx.sh I got this error:
AttributeError: module 'torch.linalg' has no attribute 'qr'
the corresponding file is fast_transformers/feature_maps/fourier_features.py, line 37.
I looked up the pytorch documentation for module torch.linalg and found that this module does not have the function qr for pytorch1.7.1 and below.
To save trouble, I didn't upgrade pytorch but downgraded pytorch-fast-transformers to version 0.3.0 which uses torch.qr instead and it worked.
So I‘m wondering if there is any version mismatch between pytorch and pytorch-fast-transformers or just because I did't follow the guide well.
Some of my conda environments are below:
CUDA version 11.0
pytorch==1.7.1
pytorch-lightning==1.1.5
pytorch-fast-transformers==0.4.0 then changed to 0.3.0
cudatoolkit==11.0.221
apex==0.1
Thank you.