bayesiains / nflows

Normalizing flows in PyTorch
MIT License
845 stars 118 forks source link

Saving models #70

Closed jankrans closed 1 year ago

jankrans commented 1 year ago

It might be something I'm looking over completely. But how can I save a trained model/weights during training? or is there an indirect way to achieve this?

Nevermind, beginner in pytroch:). Just use torch.save() & torch.load()