From 397828f880838d971ecf9fb505f75f69779ef69d Mon Sep 17 00:00:00 2001
From: Matthew Tancik <tancik@berkeley.edu>
Date: Tue, 6 Sep 2022 01:56:17 +0100
Subject: [PATCH] Pretty print configs (#343)

---
 scripts/run_train.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/scripts/run_train.py b/scripts/run_train.py
index 4e1c89b4b..d251da74d 100644
--- a/scripts/run_train.py
+++ b/scripts/run_train.py
@@ -14,6 +14,7 @@ import torch
 import torch.distributed as dist
 import torch.multiprocessing as mp
 import yaml
+from rich.console import Console
 
 from nerfactory.configs import base as cfg
 from nerfactory.configs.utils import cli_from_base_configs
@@ -192,6 +193,8 @@ def main(config: cfg.Config) -> None:
 
 
 if __name__ == "__main__":
+    console = Console(width=120)
+
     from nerfactory.configs.base_configs import base_configs
 
     instantiated_config = cli_from_base_configs(base_configs)
@@ -199,9 +202,10 @@ if __name__ == "__main__":
         logging.info(f"Loading pre-set config from: {instantiated_config.trainer.load_config}")
         instantiated_config = yaml.load(instantiated_config.trainer.load_config.read_text(), Loader=yaml.Loader)
 
-    # print and save config
-    logging.info("Printing current config setup")
-    logging.info(instantiated_config)
+    console.rule("Config")
+    console.print(instantiated_config)
+    console.rule("")
+
     assert instantiated_config.base_dir is not None
     instantiated_config.base_dir.mkdir(parents=True, exist_ok=True)
     config_yaml_path = instantiated_config.base_dir / "config.yml"