diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-09-11 22:15:44 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-09-11 22:15:44 +0200 |
| commit | cf7c8621e9c26d3dcd075ded3c91bb8ccf5a18c0 (patch) | |
| tree | a27ba6348ee28070389e276d498ca6340aa76986 | |
| parent | 56dc112cfb649217cd624b4ff305e2db83a383b7 (diff) | |
Fix logging of decoder
| -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, |