nvflare.app_opt.tracking.wandb.wandb_writer module

class WandBWriter(event_type: str = 'analytix_log_stats')[source]

Bases: LogWriter

Init FLComponent.

The FLComponent is the base class of all FL Components. (executors, controllers, responders, filters, aggregators, and widgets are all FLComponents)

FLComponents have the capability to handle and fire events and contain various methods for logging.

get_writer_name() → LogWriterName[source]

Returns “WEIGHTS_AND_BIASES”.

log(metrics: Dict[str, float], step: int | None = None)[source]

Log multiple metrics for the current run.

Parameters:

  • metrics (Dict[str, float]) – Dictionary of metric_name of type String to Float values.

  • step (int, optional) – A single integer step at which to log the specified Metrics.