From 84d01ddd251f3ba6c45e0ceaceb2a72e61efd6f7 Mon Sep 17 00:00:00 2001 From: evonneng Date: Mon, 30 May 2022 11:42:58 -0700 Subject: [PATCH] add print to beginning of train (#114) * added print unrolled config to beginning of train --- scripts/run_train.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/scripts/run_train.py b/scripts/run_train.py index 3ad5c0254..e612c3547 100644 --- a/scripts/run_train.py +++ b/scripts/run_train.py @@ -3,6 +3,7 @@ run_train_nerf.py """ import datetime import logging +import os import random import socket from datetime import timedelta @@ -15,6 +16,7 @@ import torch import torch.distributed as dist import torch.multiprocessing as mp from omegaconf import DictConfig +import yaml from radiance.nerf.trainer import Trainer from radiance.utils import comms, profiler @@ -201,6 +203,14 @@ def launch( @hydra.main(config_path="../configs", config_name="vanilla_nerf.yaml") def main(config: DictConfig): """Main function.""" + unrolled_path = os.path.join(os.getcwd(), ".hydra/config.yaml") + if os.path.exists(unrolled_path): + with open(unrolled_path, "r", encoding="utf8") as f: + unrolled_config = yaml.safe_load(f) + logger = logging.getLogger(__name__) + logger.info("Printing current config setup") + print(yaml.dump(unrolled_config, sort_keys=False, default_flow_style=False)) + launch( _train, config.machine_config.num_gpus,