開始使用免費開始

記錄評估指標

追蹤效能指標可以用來監控退化情況,你也能據此決定何時更新模型,以維持高準確度。你決定在模型完成一次評估迴圈後記錄指標。

已預先載入部分資料:

  • acceleratorAccelerator 的實例
  • eval_metric 是包含 accuracyf1 等指標的字典
  • num_epochs 是訓練輪數

本練習屬於課程

使用 PyTorch 高效訓練 AI 模型

檢視課程

練習說明

  • 呼叫方法來記錄模型的評估指標。
  • "accuracy""f1" 分數記錄為評估指標。
  • 將迴圈變數 epoch 傳入 step 參數以追蹤 epoch 編號。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

accelerator = Accelerator(project_dir=".", log_with="all")
accelerator.init_trackers("my_project")

for epoch in range(num_epochs):
    # Training loop is here
    # Evaluation loop is here
    # Call a method to log metrics
    accelerator.log({
        # Log accuracy and F1 score as metrics
        "accuracy": ____["accuracy"],
        "f1": ____["f1"],
    # Track the epoch number
    }, step=____)

accelerator.end_training()
編輯並執行程式碼