浏览代码

Log run-id in console

/develop-generalizationTraining-TrainerController
Arthur Juliani 6 年前
当前提交
fad0da30
共有 1 个文件被更改,包括 4 次插入4 次删除
  1. 8
      python/unitytrainers/trainer.py

8
python/unitytrainers/trainer.py


self.is_training and self.get_step <= self.get_max_steps):
if len(self.stats['cumulative_reward']) > 0:
mean_reward = np.mean(self.stats['cumulative_reward'])
logger.info(" {}: Step: {}. Mean Reward: {:0.3f}. Std of Reward: {:0.3f}."
.format(self.brain_name, self.get_step,
logger.info("{}: {}: Step: {}. Mean Reward: {:0.3f}. Std of Reward: {:0.3f}."
.format(self.run_id, self.brain_name, self.get_step,
logger.info(" {}: Step: {}. No episode was completed since last summary."
.format(self.brain_name, self.get_step))
logger.info("{}: {}: Step: {}. No episode was completed since last summary."
.format(self.run_id, self.brain_name, self.get_step))
summary = tf.Summary()
for key in self.stats:
if len(self.stats[key]) > 0:

正在加载...
取消
保存