1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 PyTorch 的深度学习入门
Connected

道练习

编写评估循环

在本练习中，您将编写一个评估循环来计算验证损失。评估循环与训练循环的结构相似，但不进行梯度计算或权重更新。

model、validationloader 和损失函数 criterion 已经定义，用于处理预测、数据加载和损失计算。

说明 1 / 共 2 个

undefined XP
    1
    2
  • 将模型设置为评估模式。
  • 将当前批次损失累加到变量 validation_loss。