Torch-Pruning AttributeError: Can't get attribute '__main_

AttributeError: Can't get attribute 'main' on <module 'builtins' (built-in)>

Open luoshiyong opened this issue 10 months ago • 3 comments

fine-tuning stage after yolov8-seg prune using torch-pruning, when use torch.load to load save model, the error occured

Apr 07 '24 09:04 luoshiyong

@luoshiyong hello! did you manage to fix this? I have the same error, tried everything....

Jul 06 '24 14:07 jenyaMi

@luoshiyong Hello! Same error, please let us know if you find any solution

Jul 09 '24 18:07 Ramzes30765

@luoshiyong Hello! Same error, did you fix it?

Aug 08 '24 04:08 CLemonT