浏览代码

kill trainer threads when training finishes

/develop/sac-apex
Ervin Teng 5 年前
当前提交
8d2434c2
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 3
      ml-agents/mlagents/trainers/trainer_controller.py

3
ml-agents/mlagents/trainers/trainer_controller.py


self.reset_env_if_ready(env_manager, global_step)
if self._should_save_model(global_step):
self._save_model()
# Stop advancing trainers
self.kill_trainers = True
# Final save Tensorflow model
if global_step != 0 and self.train_model:
self._save_model()

正在加载...
取消
保存