浏览代码

[skip ci] swap summary writer and step increment order

/distributed-training
Anupam Bhatnagar 5 年前
当前提交
4b960034
共有 1 个文件被更改,包括 1 次插入4 次删除
  1. 5
      ml-agents/mlagents/trainers/trainer/rl_trainer.py

5
ml-agents/mlagents/trainers/trainer/rl_trainer.py


"""
return False
# @abc.abstractmethod
self._maybe_write_summary(self.get_step + self.trainer_parameters["buffer_size"])
self._maybe_write_summary(self.get_step + self.trainer_parameters["buffer_size"])
def _increment_step(self, n_steps: int, name_behavior_id: str) -> None:
"""

:param trajectory: The Trajectory tuple containing the steps to be processed.
"""
pass
# self._maybe_write_summary(self.get_step + len(trajectory.steps))
# self._increment_step(len(trajectory.steps), trajectory.behavior_id)
def _maybe_write_summary(self, step_after_process: int) -> None:
"""

正在加载...
取消
保存