edgeai-yolov5
edgeai-yolov5 copied to clipboard
Problem with Torch and Torchvision - ModuleNotFoundError: No module named 'torch.ao.quantization'
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
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 🙂