Skip to content

Commit 2a6755c

Browse files
add run name to print
1 parent cd8eec4 commit 2a6755c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

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)