From 45dfa045d929c0b3a3f905f9c0723fe8dc809bc6 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 24 Oct 2021 00:58:54 +0200 Subject: Format training utils --- training/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'training') diff --git a/training/utils.py b/training/utils.py index 41e0a71..1996f0a 100644 --- a/training/utils.py +++ b/training/utils.py @@ -38,6 +38,7 @@ def configure_logging(config: DictConfig) -> None: def configure_callbacks(config: DictConfig,) -> List[Type[Callback]]: """Configures Lightning callbacks.""" + def load_callback(callback_config: DictConfig) -> Type[Callback]: log.info(f"Instantiating callback <{callback_config._target_}>") return hydra.utils.instantiate(callback_config) @@ -47,7 +48,7 @@ def configure_callbacks(config: DictConfig,) -> List[Type[Callback]]: for callback_config in callback_configs.values(): if callback_config.get("_target_"): callbacks.append(load_callback(callback_config)) - else: + else: callbacks += load_callbacks(callback_config) return callbacks @@ -58,6 +59,7 @@ def configure_callbacks(config: DictConfig,) -> List[Type[Callback]]: def configure_logger(config: DictConfig) -> List[Type[LightningLoggerBase]]: """Configures Lightning loggers.""" + def load_logger(logger_config: DictConfig) -> Type[LightningLoggerBase]: log.info(f"Instantiating logger <{logger_config._target_}>") return hydra.utils.instantiate(logger_config) -- cgit v1.2.3-70-g09d2