Not saving ckpt.tar.gz checkpoint about cellbender HOT 7 OPEN

same problem, not able to save check points:
Traceback (most recent call last):
File "/home2/s225139/.conda/envs/CellBender/lib/python3.8/site-packages/cellbender/remove_background/checkpoint.py", line 115, in save_checkpoint
torch.save(model_obj, filebase + '_model.torch')
File "/home2/s225139/.conda/envs/CellBender/lib/python3.8/site-packages/torch/serialization.py", line 628, in save
_save(obj, opened_zipfile, pickle_module, pickle_protocol, _disable_byteorder_record)
File "/home2/s225139/.conda/envs/CellBender/lib/python3.8/site-packages/torch/serialization.py", line 840, in _save
pickler.dump(obj)
TypeError: cannot pickle 'weakref' object

from cellbender.

No solution, but I am encountering the same error. I have tested on v0.3.2, v0.3.0 and v0.2.2. Version 0.2.2 produces expected outputs, while the more recent versions produce the errors seen above.

from cellbender.

I am also experiencing the same issue

from cellbender.

Same error.

from cellbender.

I got it working- it's a version error. I am using python 3.7.12, cellbender version 0.3.0, torch 1.13.1

from cellbender.

I got it working- it's a version error. I am using python 3.7.12, cellbender version 0.3.0, torch 1.13.1

Thank you, I will try it.

from cellbender.

I got it working- it's a version error. I am using python 3.7.12, cellbender version 0.3.0, torch 1.13.1

This combination works for me. Thanks!

from cellbender.