日志记录到 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 项目以查看微调后记录的指标。用于训练模型的配置可以在 此处 找到。