jbr-ai-labs / mamba

This code accompanies the paper "Scalable Multi-Agent Model-Based Reinforcement Learning".
MIT License
46 stars 9 forks source link

Load checkpoints and continue the training? #11

Open janetwise opened 1 year ago

janetwise commented 1 year ago

Hello,

It's usually desirable to further trainings based on previously trained models. What checkpoint models or states from previous training should be loaded to continue the training? Policy, critic, and world model? Should the replay buffer be saved and loaded too for continuous training?

Thanks!