Add save of config to runs directory

1 files changed, 7 insertions, 0 deletions

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))