登录到Comet

本深度指南将引导您了解如何在torchtune中设置日志记录到Comet。

What this deep-dive will cover

• 如何开始使用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项目以查看微调后记录的指标。 用于训练模型的配置可以在这里找到。