danijar / dreamerv2

Mastering Atari with Discrete World Models
https://danijar.com/dreamerv2
MIT License
886 stars 195 forks source link

How to save and reload trained dreamerv2 models #26

Closed Adaickalavan closed 2 years ago

Adaickalavan commented 2 years ago

1) How can I save a trained dreamerv2 model? 2) How can I reuse or load a previously trained dreamerv2 model for (i) evaluation or (ii) as a base for further training, at a later time?

danijar commented 2 years ago

The weights are checkpointed into pickle files and if you run training with the same logdir, it will automatically restore the checkpoint. Hope that helps.