edgeai-yolov5 icon indicating copy to clipboard operation
edgeai-yolov5 copied to clipboard

Problem with Torch and Torchvision - ModuleNotFoundError: No module named 'torch.ao.quantization'

Open juvinski opened this issue 2 years ago • 1 comments

Running the default command to install the requirements - pip3 install -r requirements.txt I was receiving the error ModuleNotFoundError: No module named 'torch.ao.quantization'

To solve this, you should replace - in my case I manually install pip3 install torch==1.9.0 pip3 install torchvision==0.10.0

As you can see on https://github.com/pytorch/vision/issues/5887

juvinski avatar Oct 21 '22 16:10 juvinski

Hi! I see you were trying to quantize the models.

  • Did you manage to do it?

  • If not, do you have access to the git-hash that you were using 🙂

escorciav avatar Feb 21 '23 15:02 escorciav