diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-19 21:06:22 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-19 21:06:22 +0200 |
| commit | b25c07af1986a73c2b129bfdcbefbc1dceef1885 (patch) | |
| tree | 045b9789b7f594f5f16aa683a318c4ea3cefdd63 /training/run.py | |
| parent | 53dce464914b2ce3817a7cec9441f7cfa5048248 (diff) | |
Add save of config to runs directory
Diffstat (limited to 'training/run.py')
| -rw-r--r-- | training/run.py | 1 |
1 files changed, 1 insertions, 0 deletions
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...") |