浏览代码

Print summaries at inference (#1143)

Reimplement a wrongfully reverted commit
/develop-generalizationTraining-TrainerController
GitHub 6 年前
当前提交
c8371e3b
共有 1 个文件被更改,包括 1 次插入2 次删除
  1. 3
      ml-agents/mlagents/trainers/trainer_controller.py

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


if self.train_model \
and trainer.get_step <= trainer.get_max_steps:
trainer.increment_step_and_update_last_reward()
if self.train_model:
global_step += 1
global_step += 1
if global_step % self.save_freq == 0 and global_step != 0 \
and self.train_model:
# Save Tensorflow model

正在加载...
取消
保存