Save/Load/resume training?

germain-hug / Deep-RL-Keras

Keras Implementation of popular Deep RL Algorithms (A3C, DDQN, DDPG, Dueling DDQN)

528 stars 149 forks source link

Save/Load/resume training? #10

Closed jarlva closed 5 years ago

jarlva commented 5 years ago

Hello and thanks for sharing your code! Can you please let me know if there is a way to save model state/weights in order to test later or resume training?

germain-hug commented 5 years ago

Hi and thanks for your interest! I just modified main.py to save weights when done training. I also added a separate script (load_and_run.py) for visualization after training.

jarlva commented 5 years ago

Great! I'll give it a shot