撰寫評估迴圈
在這個練習中,你將撰寫一個評估迴圈來計算驗證損失。評估迴圈的結構與訓練迴圈相似,但不進行梯度計算或權重更新。
model、validationloader,以及損失函式 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 += ____