diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-08-25 23:19:39 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-08-25 23:19:39 +0200 |
commit | 6968572c1a21394b88a29f675b17b9698784a898 (patch) | |
tree | d89d1c5c2ec331d38dcb5b6a2dbbd72c9e355b8a /training/run.py | |
parent | 49ca6ade1a19f7f9c702171537fe4be0dfcda66d (diff) |
Update training stuff
Diffstat (limited to 'training/run.py')
-rw-r--r-- | training/run.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/training/run.py b/training/run.py index 288a1ef..cffc3ae 100644 --- a/training/run.py +++ b/training/run.py @@ -11,7 +11,7 @@ from pytorch_lightning import ( seed_everything, Trainer, ) -from pytorch_lightning.loggers import LightningLoggerBase +from pytorch_lightning.loggers import Logger from torch import nn from torchinfo import summary import utils @@ -55,7 +55,7 @@ def run(config: DictConfig) -> Optional[float]: # Load callback and logger. callbacks: List[Type[Callback]] = utils.configure_callbacks(config) - logger: List[Type[LightningLoggerBase]] = utils.configure_logger(config) + logger: List[Type[Logger]] = utils.configure_logger(config) log.info(f"Instantiating trainer <{config.trainer._target_}>") trainer: Trainer = hydra.utils.instantiate( |