`torch_mist.utils.logging.logger.wandb`

Module Contents

Classes

class torch_mist.utils.logging.logger.wandb.WandbLogger(project: str, log_dir: str | None = None, log_every: int = 10)

Bases: torch_mist.utils.logging.logger.base.Logger

add_config(config: Dict[str, Any])

_log(data: Any, name: str, iteration: int, epoch: int, split: str)

get_log() → wandb.wandb_sdk.wandb_run.Run

_reset_log()

save_model(model: torch.nn.Module, name: str, artifact_name: str | None = None)