From b25c07af1986a73c2b129bfdcbefbc1dceef1885 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
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(+)

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