Error while replicating the flow suggested in Readme.md

Question

Error while replicating the flow suggested in Readme.md

max190 opened this issue a year ago · comments

I git cloned the repository and tried to run the flow suggested in the README.md and run the compress.py file, but I get an error as "Start Pruning" step begins.
Attaching the image to show the corresponding error.
Error: "RuntimeError: result type Float can't be cast to the desired output type long int"
Command: python compress.py --model test1 --dataset COCO --data coco.yaml --batch 64 --weights yolov5s.pt --initial_rate 0.06 --initial_thres 6. --topk 0.8 --exp --device 0

Hank Ye · Answer 1 · Tue May 02 2023 23:28:27 GMT+0800 (China Standard Time)

Hi, @max190. You can refer to this issue for more help. To address this problem, you may either degrade the pytorch version or upgrade the yolov5 code.