From d301c02b2d392c65a36dc81e2103c0925a37f2e2 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 27 Jun 2021 20:26:04 +0200 Subject: Renaming --- training/run_experiment.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'training') 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, -- cgit v1.2.3-70-g09d2