jenyaMi

AttributeError: Can't get attribute '__main__' on <module 'builtins' (built-in)> while loading the pruned model

5

I use yolov8 pruning code, to be more precisely the updated version for the new version of ultralytics: https://github.com/chbw818/yolov8-prune-using-torch-pruning-/tree/main I am struggling with loading the pruned model... AttributeError: Can't get...