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)