flow-project / flow

Computational framework for reinforcement learning in traffic control
MIT License
1.04k stars 369 forks source link

Finding best checkpoint among all trained checkpoints #1044

Open yulinzhang opened 3 years ago

yulinzhang commented 3 years ago

Hi,

After the model is trained, I use tensorboard to find that the reward mean goes up and down. How do I automatically identify the checkpoint (number) that gives the peak in the reward mean/max/min?

With this peak checkpoint, I can further train other models starting from this checkpoint.

Best, Yulin