Skip to content

Commit afa0cb0

Browse files
fix: include day count in long training log durations (#210)
--------- Co-authored-by: ClemensSchwarke <clemens.schwarke@gmail.com>
1 parent 3b0646a commit afa0cb0

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

CONTRIBUTORS.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ Please keep the lists sorted alphabetically.
3636
* Lorenzo Terenzi
3737
* Marko Bjelonic
3838
* Markus Portugall
39+
* Masahiro Kubota
3940
* Matthijs van der Boon
4041
* Özhan Özen
4142
* Pascal Roth

rsl_rl/utils/logger.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@
66

77
from __future__ import annotations
88

9+
import datetime
910
import git
1011
import os
1112
import pathlib
1213
import statistics
13-
import time
1414
import torch
1515
import warnings
1616
from collections import deque
@@ -274,8 +274,8 @@ def log(
274274
log_string += (
275275
f"""{"-" * width}\n"""
276276
f"""{"Iteration time:":>{pad}} {iteration_time:.2f}s\n"""
277-
f"""{"Time elapsed:":>{pad}} {time.strftime("%H:%M:%S", time.gmtime(self.tot_time))}\n"""
278-
f"""{"ETA:":>{pad}} {time.strftime("%H:%M:%S", time.gmtime(eta))}\n"""
277+
f"""{"Time elapsed:":>{pad}} {datetime.timedelta(seconds=int(self.tot_time))}\n"""
278+
f"""{"ETA:":>{pad}} {datetime.timedelta(seconds=int(eta))}\n"""
279279
)
280280
print(log_string)
281281

0 commit comments

Comments
 (0)