From cf7c8621e9c26d3dcd075ded3c91bb8ccf5a18c0 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 11 Sep 2023 22:15:44 +0200 Subject: Fix logging of decoder --- training/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training') diff --git a/training/run.py b/training/run.py index cffc3ae..b8f2178 100644 --- a/training/run.py +++ b/training/run.py @@ -34,7 +34,7 @@ def run(config: DictConfig) -> Optional[float]: log.info(f"Instantiating criterion <{config.criterion._target_}>") loss_fn: Type[nn.Module] = hydra.utils.instantiate(config.criterion) - log.info(f"Instantiating decoder <{config.criterion._target_}>") + log.info(f"Instantiating decoder <{config.decoder._target_}>") decoder: Type[Callable] = hydra.utils.instantiate( config.decoder, network=network, -- cgit v1.2.3-70-g09d2