在 GitHub 上编辑

Fast.ai

DVCLive 可以为你的 Fast.ai 项目添加实验跟踪功能。

用法

DVCLiveCallback 添加到传递给 Learner 的回调列表中:

from dvclive.fastai import DVCLiveCallback

...

learn = tabular_learner(data_loader, metrics=accuracy)
learn.fit_one_cycle(
    n_epoch=2,
    cbs=[DVCLiveCallback()])

每个指标将被记录到:

{Live.plots_dir}/metrics/{split}/{metric}.tsv

其中:

  • {Live.plots_dir}Live 中定义。
  • {split} 可以是 traineval
  • {metric} 是框架提供的名称。

参数

  • live -(默认为 None)- 可选的 Live 实例。若为 None,将使用 **kwargs 创建一个新实例。

  • **kwargs - 任何其他参数都将用于实例化一个新的 Live 实例。若使用了 live,这些参数将被忽略。

示例

  • 使用 live 传入一个已有的 Live 实例。
from dvclive import Live
from dvclive.fastai import DVCLiveCallback

with Live("custom_dir") as live:
    learn = tabular_learner(data_loader, metrics=accuracy)
    learn.fit_one_cycle(
      n_epoch=2,
      cbs=[DVCLiveCallback(live=live)])

    # Log additional metrics after training
    live.log_metric("summary_metric", 1.0, plot=False)
  • 使用 **kwargs 自定义新的 Live 实例。
learn.fit_one_cycle(
  n_epoch=2,
  cbs=[DVCLiveCallback(dir="custom_dir")])
内容

🐛 发现问题?告诉我们!或者修复它:

在 GitHub 上编辑

有疑问?加入我们的聊天,我们会为您提供帮助:

Discord 聊天