WANDB结果跟踪器

class WANDBResultTracker(project: str, offline: bool = False, **kwargs)[源代码]

基础类:ResultTracker

一个用于Weights and Biases的跟踪器。

请注意,您必须事先执行wandb登录。

通过WANDB初始化结果跟踪。

Parameters:
  • 项目 (str) – 您的WANDB登录有权访问的项目名称。

  • offline (bool) – 是否在离线模式下运行,即不与wandb服务器同步。

  • kwargs – 传递给 wandb.init() 的额外关键字参数。

Raises:

ValueError – 如果项目名称为None

方法总结

end_run([success])

结束运行。

log_metrics(metrics[, step, prefix])

将指标记录到结果存储中。

log_params(params[, prefix])

将日志参数记录到结果存储中。

start_run([run_name])

使用可选名称启动运行。

方法文档

end_run(success: bool = True) None[source]

结束运行。

必须在实验结束后调用。

Parameters:

success (bool) – 可用于表示运行失败。可能会被忽略。

Return type:

log_metrics(metrics: Mapping[str, float], step: int | None = None, prefix: str | None = None) None[source]

将指标记录到结果存储中。

Parameters:
  • metrics (Mapping[str, float]) – 要记录的指标。

  • step (int | None) – 一个可选的步骤,用于附加指标(例如,epoch)。

  • prefix (str | None) – 一个可选的前缀,用于在指标中的每个键前添加。

Return type:

log_params(params: Mapping[str, Any], prefix: str | None = None) None[来源]

将日志参数记录到结果存储中。

Parameters:
Return type:

start_run(run_name: str | None = None) None[source]

使用可选名称启动运行。

Parameters:

run_name (str | None)

Return type: