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

VainF / Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs

https://arxiv.org/abs/2301.12900

MIT License

2.44k stars 308 forks source link

Open luoshiyong opened 2 months ago

luoshiyong commented 2 months ago

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