FLAP icon indicating copy to clipboard operation
FLAP copied to clipboard

Published 20 hours ago • verified publisher icon CASIA-IVA-Lab

AttributeError: Can't pickle local object 'add_hook_to_module.<locals>.new_forward'

Open shwu-nyunai opened this issue 10 months ago • 1 comments

Traceback (most recent call last): File "/home/shwu/LABS/FLAP/main.py", line 147, in main() File "/home/shwu/LABS/FLAP/main.py", line 141, in main torch.save(model, f'{args.save_model}/pruned_model.pt') File "/home/shwu/LABS/FLAP/venv/lib/python3.10/site-packages/torch/serialization.py", line 629, in save _save(obj, opened_zipfile, pickle_module, pickle_protocol, _disable_byteorder_record) File "/home/shwu/LABS/FLAP/venv/lib/python3.10/site-packages/torch/serialization.py", line 841, in _save pickler.dump(obj) AttributeError: Can't pickle local object 'add_hook_to_module..new_forward'

this occurs after pruning on multiple gpus and saving with torch.save

shwu-nyunai avatar Apr 09 '24 12:04 shwu-nyunai

Hello, I have the same problem, did you solve it?

NEUBuffett avatar Jul 18 '24 06:07 NEUBuffett