WandbLogger

class unike.utils.WandbLogger(endpoint: Literal['wandb', 'swanlab'] = 'wandb')

使用 Weights and Biases 记录实验结果。

__init__(endpoint: Literal['wandb', 'swanlab'] = 'wandb')

创建 WandbLogger 对象。

参数:

endpoint (Literal['wandb', 'swanlab']) – 使用 wandb 还是 swanlab 记录实验结果

__weakref__

list of weak references to the object (if defined)

finish()

关闭日志

log(*args, **kwargs)

记录日志

set_config(project: str, name: str, config: dict[str, Any] | None = None, offline: bool = False) → WandbLogger

设置项目名称和配置。

参数:

• project (str) – wandb 的项目名称

• name (str) – wandb 的 run name

• config (dict[str, Any] | None) – wandb 的项目配置如超参数。

• offline (bool) – 是否离线模式