diff options
Diffstat (limited to 'training/utils.py')
| -rw-r--r-- | training/utils.py | 7 |
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)) |