Open cxq1 opened 2 years ago
It is possibly caused by incompatible PyTorch version. Which PyTorch version do you use?
It is possibly caused by incompatible PyTorch version. Which PyTorch version do you use?
'1.10.1
We use PyTorch 1.7 to run all the experiments. I think PT 1.7.0-1.8.0 will be fine. PT 1.9+ are untested.
@DeppMeng Is there an extra 's' on the line 30 of models/attention.py ? _LinearWithBiass or _LinearWithBias ?
@Li-Qingyun Thank you for pointing out, the issue is fixed.
cannot import name '_LinearWithBias' from 'torch.nn.modules.linear'