search

ministry-of-silly-code / experiment_buddy

GNU Affero General Public License v3.0
9 stars 5 forks source link

Save jitted models and other pickable objects #109

Open d3sm0 opened 2 years ago

d3sm0 commented 2 years ago

As of now we have to use torch.save to save objects. it would be nice to save models with another method like torch.jit or pickle for general files like a log file.

I can proceed with a pr if you like.