diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-07 22:26:29 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-07 22:26:29 +0200 |
| commit | f4629a0d4149d5870c9fd8ce83ff5d391bd7ddd3 (patch) | |
| tree | 3d4364a53a05a9b019b3af430d3092d434e2be1c /training | |
| parent | 8afa8e1c6e9623b0dea86236da04b2b4173e9443 (diff) | |
Typing fixed, tasks removed from noxfile
Diffstat (limited to 'training')
| -rw-r--r-- | training/run_experiment.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/run_experiment.py b/training/run_experiment.py index 1fb4bc2..491112c 100644 --- a/training/run_experiment.py +++ b/training/run_experiment.py @@ -69,7 +69,7 @@ def _configure_callbacks( def _configure_logger( network: Type[nn.Module], args: Dict, use_wandb: bool -) -> pl.loggers.WandbLogger: +) -> Type[pl.loggers.LightningLoggerBase]: """Configures lightning logger.""" if use_wandb: pl_logger = pl.loggers.WandbLogger() |