Improved model saving with Checkpoint system

pranz24 / pytorch-soft-actor-critic

PyTorch implementation of soft actor critic

MIT License

810 stars 180 forks source link

Improved model saving with Checkpoint system #37

Closed shnippi closed 3 years ago

shnippi commented 3 years ago

Improved saving for the SAC by making checkpoints files. These enable to save your SAC to checkpoints files and continue your training from there. This was previously not possible.

pranz24 commented 3 years ago

Looks good to me!

Thanks!