diff options
| -rw-r--r-- | training/run.py | 2 |
1 files changed, 1 insertions, 1 deletions
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, |