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