diff --git a/CONTRIBUTORS.md b/CONTRIBUTORS.md index 00692ba7..bafc69ed 100644 --- a/CONTRIBUTORS.md +++ b/CONTRIBUTORS.md @@ -37,5 +37,6 @@ Please keep the lists sorted alphabetically. * Matthijs van der Boon * Özhan Özen * Pascal Roth +* Shaoshu Su * Zhang Chong * Ziqi Fan diff --git a/rsl_rl/utils/logger.py b/rsl_rl/utils/logger.py index c53e44c6..2dc29231 100644 --- a/rsl_rl/utils/logger.py +++ b/rsl_rl/utils/logger.py @@ -182,6 +182,10 @@ def log( log_string = f"""{"#" * width}\n""" log_string += f"""\033[1m{f" Learning iteration {it}/{total_it} ".center(width)}\033[0m \n\n""" + # Print run name if provided + run_name = self.cfg.get("run_name") + log_string += f"""{"Run name:":>{pad}} {run_name}\n""" if run_name else "" + # Print performance log_string += ( f"""{"Total steps:":>{pad}} {self.tot_timesteps} \n"""