cellflow.training.WandbLogger

class cellflow.training.WandbLogger(project, out_dir, config, **kwargs)[source]

Callback to log data to Weights and Biases

Parameters:

  • project (str) – The project name in wandb

  • out_dir (str) – The output directory to save the logs

  • config (dict[str, Any]) – The configuration to log

  • **kwargs – Additional keyword arguments to pass to wandb.init()

Returns:

None

Methods

on_log_iteration(dict_to_log, **_)

Called at each validation/log iteration to log data to WandB

on_train_begin()

Called at the beginning of training to initiate WandB logging

on_train_end(dict_to_log)

Called at the end of training to log data to WandB