megatron.wandb_logger.WandbTBShim

class megatron.wandb_logger.WandbTBShim(config: WandBConfig)

Bases: object

Shim class that holds the configuration we want the wandb wrapper to use (e.g. to control sampling, delay upload etc) and that translates the API