microsoft / CvT

This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.
MIT License
533 stars 120 forks source link

recommended torch version may be wrong #22

Open RylonW opened 1 year ago

RylonW commented 1 year ago

After installing torch 1.7.1, I got an ERROR: ModuleNotFoundError: No module named 'torch.fx What I find on stackoverflow is that torch.fx was added in PyTorch 1.8.0., so may be recommended version is wrong?