summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-27 20:26:04 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-27 20:26:04 +0200
commitd301c02b2d392c65a36dc81e2103c0925a37f2e2 (patch)
tree531169147eb0df43b089f22e001b8b602157fc35 /training
parent442eac315e4b8be19adab80fb7332d29f68c077c (diff)
Renaming
Diffstat (limited to 'training')
-rw-r--r--training/run_experiment.py6
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,