Cannot import name 'QuantStub' from 'torch.ao.quantization'

what’s your torchvision version? perhaps need update or downgrade it…