rlcode / reinforcement-learning

Minimal and Clean Reinforcement Learning Examples

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Saved model usage

akileshbadrinaaraayanan opened this issue · comments

Could you please provide examples on how to use the saved model (.h5 files) at test time for Grid world, Cartpole environments?

Thanks,
Akilesh

Just wondering if you saw the functions "agent.model.save_weights" and "agent.model.load_weights" in CartPole DDQN implementation code.
https://github.com/rlcode/reinforcement-learning/blob/master/2-cartpole/2-double-dqn/cartpole_ddqn.py

You can use saved model by changing self.load_model from False to True. Take care about the name of saved model. Thanks