How to train a new model from stage1_1.pth or stage1_2.pth ?

Acmece / rl-collision-avoidance

Implementation of the paper "Towards Optimally Decentralized Multi-Robot Collision Avoidance via Deep Reinforcement Learning"

https://arxiv.org/abs/1709.10082

326 stars 92 forks source link

How to train a new model from stage1_1.pth or stage1_2.pth ? #4

Closed utamizu closed 4 years ago

utamizu commented 4 years ago

Hi. I am confused about how to train such a stage2.pth model from stage1_1.pth or stage1_2.pth . I'm currently loading stage1_1.pth and learning stage2, but I can't generate a model like stage2.pth. Do you have any tips?