How to save scgen trained model?

theislab / scgen

Single cell perturbation prediction

https://scgen.readthedocs.io

GNU General Public License v3.0

260 stars 52 forks source link

How to save scgen trained model? #36

Closed levinhein closed 3 years ago

levinhein commented 3 years ago

Hello. How can I save the scgen objects and scgen model? I have experienced running scgen 30hrs for three times because jupyter notebook crashed after training and during prediction.

I tried using pickle module to save the model (hoping that I could just load it again anytime instead or re-running scgen many times) but it doesn't seem to work and reverts an error. Advise, please. Thank you very much!

M0hammadL commented 3 years ago

We have released a new PyTorch version of the code. Please use this version.

Best Mo