Comet.ml

与 Comet.ml 集成。

注册

  1. 创建账户：comet.ml/signup。
  2. 将 API 密钥导出到环境变量（更多帮助请参见此处）。在您的终端中运行
export COMET_API_KEY='YOUR_LONG_API_TOKEN'

或将其包含在您的 ./comet.config 文件中（推荐）。更多帮助请参见此处。

安装

  1. 您需要安装 neptune-client。在您的终端中运行
pip install comet_ml

或（使用 conda 的替代安装方法）。在您的终端中运行

conda install -c anaconda -c conda-forge -c comet_ml comet_ml

如何使用？

关键是在创建 Learner() 之前创建回调 CometMLCallback，如下所示

from fastai.callback.comet import CometMLCallback

comet_ml_callback = CometCallback('PROJECT_NAME')  # specify project

learn = Learner(dls, model,
                cbs=comet_ml_callback
                )

learn.fit_one_cycle(1)

源

CometCallback

 CometCallback (project_name, log_model_weights=True)

将损失、指标、模型权重、模型架构摘要记录到 neptune