From b25c07af1986a73c2b129bfdcbefbc1dceef1885 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 19 Sep 2021 21:06:22 +0200 Subject: Add save of config to runs directory --- training/run.py | 1 + training/utils.py | 7 +++++++ 2 files changed, 8 insertions(+) (limited to 'training') diff --git a/training/run.py b/training/run.py index 0cf52e3..ddae2b9 100644 --- a/training/run.py +++ b/training/run.py @@ -67,6 +67,7 @@ def run(config: DictConfig) -> Optional[float]: # Log hyperparameters log.info("Logging hyperparameters") utils.log_hyperparameters(config=config, model=model, trainer=trainer) + utils.save_config(config) if config.debug: log.info("Fast development run...") diff --git a/training/utils.py b/training/utils.py index d23396e..6f57e4d 100644 --- a/training/utils.py +++ b/training/utils.py @@ -17,7 +17,14 @@ from tqdm import tqdm import wandb +def save_config(config: DictConfig) -> None: + """Save config to experiment directory.""" + with open("config.yaml", "w") as f: + OmegaConf.save(config, f=f) + + def print_config(config: DictConfig) -> None: + """Prints config.""" print(OmegaConf.to_yaml(config)) -- cgit v1.2.3-70-g09d2