diff options
Diffstat (limited to 'training/utils.py')
| -rw-r--r-- | training/utils.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/training/utils.py b/training/utils.py index d1801a7..f0a0d4d 100644 --- a/training/utils.py +++ b/training/utils.py @@ -1,6 +1,6 @@ """Util functions for training with hydra and pytorch lightning.""" -from typing import List, Type import warnings +from typing import List, Type import hydra from loguru import logger as log @@ -14,6 +14,7 @@ from pytorch_lightning.loggers import Logger from pytorch_lightning.loggers.wandb import WandbLogger from pytorch_lightning.utilities import rank_zero_only from tqdm import tqdm + import wandb |