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