Logging to Comet

本深入指南将引导您了解如何在 torchtune 中设置日志记录以集成 Comet。

本深入探讨将涵盖以下内容

• 如何开始使用 Comet

• 如何使用 CometLogger

• 如何将配置、指标和模型检查点记录到 Comet

torchtune 支持将您的训练运行日志记录到 Comet。 下面的截图展示了一个来自 torchtune 微调运行的 Comet 工作区示例。

注意

您需要安装 comet_ml 包才能使用此功能。 您可以使用 pip 安装它：

pip install comet_ml

您可能还需要登录 Comet 以开始记录数据。您可以通过命令行执行以下操作：

comet login

指标记录器

您唯一需要做的更改是将指标记录器添加到您的配置中。Comet 将为您记录指标和模型检查点。

# enable logging to the built-in CometLogger
metric_logger:
  _component_: torchtune.training.metric_logging.CometLogger
  # the Comet project to log to
  project: comet-examples-torchtune
  experiment_name: my-experiment-name

我们自动从您运行的配方中获取配置并将其记录到Comet。您可以在Comet的超参数选项卡中找到它，实际文件在Assets & Artifacts选项卡中。

注意

点击此示例 Comet 项目以查看微调后的记录指标。 用于训练模型的配置可以在 此处 找到。