File size: 331 Bytes
5910540 | 1 2 3 4 5 6 7 8 9 10 11 | from pytorch_lightning.loggers import WandbLogger
def get_wandb_logger(loggers):
"""Gets the wandb logger if it is the list of loggers otherwise returns None."""
wandb_logger = None
for logger in loggers:
if isinstance(logger, WandbLogger):
wandb_logger = logger
return wandb_logger
|