ModuleNotFoundError: No module named ‘torch.ao‘ 报错解决

问题

ModuleNotFoundError: No module named ‘torch.ao’

来源

用vformer库跑视频transformer，在

from torch.ao.quantization import QuantStub, DeQuantStub

处报错，没有torch.ao

原因

torch版本太新

解决

把

torch.ao.quantization

改为

torch.quantization

即可运行