浏览代码

TensorBoard Lesson -> Lesson Number (#4347)

/MLA-1734-demo-provider
GitHub 4 年前
当前提交
a74c7bc5
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 2
      ml-agents/mlagents/trainers/stats.py
  2. 2
      ml-agents/mlagents/trainers/trainer_controller.py

2
ml-agents/mlagents/trainers/stats.py


class GaugeWriter(StatsWriter):
"""
Write all stats that we recieve to the timer gauges, so we can track them offline easily
Write all stats that we receive to the timer gauges, so we can track them offline easily
"""
@staticmethod

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


) in self.param_manager.get_current_lesson_number().items():
for trainer in self.trainers.values():
trainer.stats_reporter.set_stat(
f"Environment/Lesson/{param_name}", lesson_number
f"Environment/Lesson Number/{param_name}", lesson_number
)
for trainer in self.trainers.values():

正在加载...
取消
保存