记录到 Comet

本篇深入探讨将指导你如何在 torchtune 中设置 Comet 的日志记录。

本次深入探讨将涵盖什么

• 如何开始使用 Comet

• 如何使用

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

torchtune 支持将您的训练运行记录到 Comet。 下面的屏幕截图中可以看到 torchtune 微调运行中的 Comet 工作区示例。

注意

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

pip install comet_ml

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

comet login

公制记录器

您需要进行的唯一更改是将 metric logger 添加到您的配置中。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 Hyperparameters （彗星超参数） 选项卡中找到它，并在选项卡中找到实际文件。Assets & Artifacts

注意

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