From 4ca2c2725ca9f8a61b3dc61cac5ad3ac7dec87a0 Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Sun, 27 Nov 2022 21:06:56 -0800 Subject: [PATCH] Cleanup train cli logs --- nerfstudio/engine/trainer.py | 6 +++--- nerfstudio/utils/writer.py | 8 +++----- 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py index d35ebfeee4..ae9da1ec2c 100644 --- a/nerfstudio/engine/trainer.py +++ b/nerfstudio/engine/trainer.py @@ -198,10 +198,10 @@ def _check_viewer_warnings(self) -> None: """Helper to print out any warnings regarding the way the viewer/loggers are enabled""" if self.config.is_viewer_enabled(): string = ( - "[WARNING] Not running eval iterations since only viewer is enabled." - " Use `--vis wandb` or `--vis tensorboard` to run with eval instead." + "[NOTE] Not running eval iterations since only viewer is enabled." + " Use [yellow]--vis wandb[/yellow] or [yellow]--vis tensorboard[/yellow] to run with eval instead." ) - CONSOLE.print(f"[bold red]{string}") + CONSOLE.print(f"{string}") @check_viewer_enabled def _init_viewer_state(self) -> None: diff --git a/nerfstudio/utils/writer.py b/nerfstudio/utils/writer.py index e1ed07057d..76d7a3aa91 100644 --- a/nerfstudio/utils/writer.py +++ b/nerfstudio/utils/writer.py @@ -208,8 +208,8 @@ def setup_event_writer(config: cfg.Config, log_dir: Path) -> None: if using_event_writer: string = f"logging events to: {log_dir}" else: - string = "disabled tensorboard/wandb event writers" - CONSOLE.print(f"[bold red]{string}") + string = "Disabled tensorboard/wandb event writers" + CONSOLE.print(f"[bold yellow]{string}") class Writer: @@ -387,7 +387,7 @@ def write_stats_log(self, step: int) -> None: if not self.has_printed and self.config.max_log_size: CONSOLE.log( f"Printing max of {self.config.max_log_size} lines. " - "Set flag `--logging.local-writer.max-log-size=0` " + "Set flag รงรงรงรง[yellow]--logging.local-writer.max-log-size=0[/yellow] " "to disable line wrapping." ) latest_map, new_key = self._consolidate_events() @@ -430,10 +430,8 @@ def _update_header(self, latest_map, new_key): self.past_mssgs[0] = mssg self.past_mssgs[1] = "-" * len(mssg) if full_log_cond or not self.has_printed: - print("testeing testing!") print(mssg) print("-" * len(mssg)) - # self.has_printed = True def _print_stats(self, latest_map, padding=" "): """helper to print out the stats in a readable format