nip.stat_logger.WandbStatLogger

nip.stat_logger.WandbStatLogger#

class nip.stat_logger.WandbStatLogger(wandb_run: Run)[source]#

A logger that logs statistics to Weights & Biases.

Methods Summary

__init__(wandb_run)

log(to_log[, step])

Log some statistics with Weights & Biases.

Methods

__init__(wandb_run: Run)[source]#
log(to_log: dict, step: int | None = None)[source]#

Log some statistics with Weights & Biases.

Parameters:
  • to_log (dict) – The statistics to log.

  • step (int, optional) – The step at which the statistics are to be logged.