Error while replicating the flow suggested in Readme.md

[max190]

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

[HankYe]

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.