Skip to content

Commit 1e49fd0

Browse files
authored
Account for start_iter when computing ETA (#29)
1 parent bee0d14 commit 1e49fd0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

rsl_rl/runners/on_policy_runner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -351,8 +351,8 @@ def log(self, locs: dict, width: int = 80, pad: int = 35):
351351
f"""{'Total timesteps:':>{pad}} {self.tot_timesteps}\n"""
352352
f"""{'Iteration time:':>{pad}} {iteration_time:.2f}s\n"""
353353
f"""{'Total time:':>{pad}} {self.tot_time:.2f}s\n"""
354-
f"""{'ETA:':>{pad}} {self.tot_time / (locs['it'] + 1) * (
355-
locs['num_learning_iterations'] - locs['it']):.1f}s\n"""
354+
f"""{'ETA:':>{pad}} {self.tot_time / (locs['it'] - locs['start_iter'] + 1) * (
355+
locs['start_iter'] + locs['num_learning_iterations'] - locs['it']):.1f}s\n"""
356356
)
357357
print(log_string)
358358

0 commit comments

Comments
 (0)