It seems that the file prune.py can't be found in github warehouse of yolov5. Maybe you customized it? Look at the official train.py code, which mentions that external use can adjust the running function. Could it be a mistake with val.py? This is pure speculation.