Skip to content

Commit 23a5b2a

Browse files
Include run_name in training logs for better run traceability (#101)
--------- Co-authored-by: ClemensSchwarke <clemens.schwarke@gmail.com>
1 parent 17bb7c0 commit 23a5b2a

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

CONTRIBUTORS.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,5 +37,6 @@ Please keep the lists sorted alphabetically.
3737
* Matthijs van der Boon
3838
* Özhan Özen
3939
* Pascal Roth
40+
* Shaoshu Su
4041
* Zhang Chong
4142
* Ziqi Fan

rsl_rl/utils/logger.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,10 @@ def log(
182182
log_string = f"""{"#" * width}\n"""
183183
log_string += f"""\033[1m{f" Learning iteration {it}/{total_it} ".center(width)}\033[0m \n\n"""
184184

185+
# Print run name if provided
186+
run_name = self.cfg.get("run_name")
187+
log_string += f"""{"Run name:":>{pad}} {run_name}\n""" if run_name else ""
188+
185189
# Print performance
186190
log_string += (
187191
f"""{"Total steps:":>{pad}} {self.tot_timesteps} \n"""

0 commit comments

Comments
 (0)