nemo_automodel.loggers.wandb_utils

Module Contents

Functions

suppress_wandb_log_messages

Patches wandb logger to suppress upload messages.

API

nemo_automodel.loggers.wandb_utils.suppress_wandb_log_messages()

Patches wandb logger to suppress upload messages.

These occur usually on KeyboardInterrupt or program crash.

To print the log url: run = wandb.init(…) print(run.url)