diff options
Diffstat (limited to 'training')
| -rw-r--r-- | training/run_experiment.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/training/run_experiment.py b/training/run_experiment.py index 607c3ce..def1e77 100644 --- a/training/run_experiment.py +++ b/training/run_experiment.py @@ -2,7 +2,7 @@ from datetime import datetime import importlib from pathlib import Path -from typing import Dict, List, Optional, Type +from typing import List, Optional, Type import hydra from loguru import logger @@ -27,7 +27,7 @@ def _create_experiment_dir(config: DictConfig) -> Path: return log_dir -def save_config(config: DictConfig, log_dir: Path) -> None: +def _save_config(config: DictConfig, log_dir: Path) -> None: """Saves config to log directory.""" with (log_dir / "config.yaml").open("r") as f: OmegaConf.save(config=config, f=f) @@ -132,7 +132,7 @@ def run(config: DictConfig) -> None: lit_model = _load_lit_model(lit_model_class, network, config) # Save config to experiment dir. - save_config(config, log_dir) + _save_config(config, log_dir) trainer = pl.Trainer( **config.trainer.args, |