開始使用免費開始

撰寫評估迴圈

在這個練習中,你將撰寫一個評估迴圈來計算驗證損失。評估迴圈的結構與訓練迴圈相似,但不進行梯度計算或權重更新

modelvalidationloader,以及損失函式 criterion 已經定義好,分別用來處理預測、資料載入與損失計算。

本練習屬於課程

使用 PyTorch 的深度學習入門

檢視課程

動手互動練習

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

# Set the model to evaluation mode
____
validation_loss = 0.0

with torch.no_grad():
  
  for features, labels in validationloader:
    
      outputs = model(features)
      loss = criterion(outputs, labels)
      
      # Sum the current loss to the validation_loss variable
      validation_loss += ____
編輯並執行程式碼