- class finetuner.tuner.callback.wandb_logger.WandBLogger(**kwargs)¶
Weights & Biases logger to log metrics for training and validation. To use this logger, make sure to have a WandB account created, install the WandB client (which you can do using
pip install wandb) and (using
wandb loginor by setting the API key as environmental variable).
Initialize the WandB logger :param kwargs: Keyword arguments that are passed to
Called at the end of a training batch, after the backward pass.
Called at the end of an evaluation batch.
Called at the end of the evaluation.
Called at the end of an epoch, after both training and validation (or just training if no validaton is provided).
Called at the end of the
fitmethod call, after finishing all the epochs.