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

[luoshiyong]

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

[jenyaMi]

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

[Ramzes30765]

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

[CLemonT]

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