I trained a model in the pybullet_ant environment with
python -m pybullet_envs.agents.train_ppo --config=pybullet_ant --logdir=ant
The trained model is now stored in ant/xxxxx (where xxxxx is a time-stamped subdirectory). Can I force train_ppo to load the trained model in a new training session? So I would continue training it, instead of starting from scratch?
Thank you.
How to continue training a model with train_ppo ?
-
- Posts: 4
- Joined: Tue Jul 24, 2018 7:45 pm
-
- Site Admin
- Posts: 4221
- Joined: Sun Jun 26, 2005 6:43 pm
- Location: California, USA
Re: How to continue training a model with train_ppo ?
If you want to resume a previously started run, add the --timestamp=<time> flag to the last command and provide the timestamp in the directory name of your run.
See also the more new version of PPO for more info on the --timestamp option.
https://github.com/google-research/batch-ppo
(the new version may not be compatible with PyBullet though, may need some work)
See also the more new version of PPO for more info on the --timestamp option.
https://github.com/google-research/batch-ppo
(the new version may not be compatible with PyBullet though, may need some work)