From 1a2e46c51cf8b489bf293b543bf4d80dbfb8052f Mon Sep 17 00:00:00 2001 From: Peter Mitrano Date: Mon, 17 Jun 2024 20:53:53 +0000 Subject: [PATCH] account for start_iter when computing ETA --- rsl_rl/runners/on_policy_runner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rsl_rl/runners/on_policy_runner.py b/rsl_rl/runners/on_policy_runner.py index 9e0a459..6aa5344 100644 --- a/rsl_rl/runners/on_policy_runner.py +++ b/rsl_rl/runners/on_policy_runner.py @@ -245,8 +245,8 @@ def log(self, locs: dict, width: int = 80, pad: int = 35): f"""{'Total timesteps:':>{pad}} {self.tot_timesteps}\n""" f"""{'Iteration time:':>{pad}} {iteration_time:.2f}s\n""" f"""{'Total time:':>{pad}} {self.tot_time:.2f}s\n""" - f"""{'ETA:':>{pad}} {self.tot_time / (locs['it'] + 1) * ( - locs['num_learning_iterations'] - locs['it']):.1f}s\n""" + f"""{'ETA:':>{pad}} {self.tot_time / (locs['it'] - locs['start_iter'] + 1) * ( + locs['start_iter'] + locs['num_learning_iterations'] - locs['it']):.1f}s\n""" ) print(log_string)