danijar / dreamerv2

Mastering Atari with Discrete World Models

Home Page:https://danijar.com/dreamerv2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to save and reload trained dreamerv2 models

Adaickalavan opened this issue · comments

commented
  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?

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.