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

道练习

尝试使用 dropout

Dropout 通过在训练过程中随机将部分输出值置零来帮助防止过拟合。在本练习中，您将构建一个带有 dropout 的简单神经网络，并观察它在训练模式与评估模式下的表现。

torch.nn 包已作为 nn 预加载，features 也已为您定义。

说明 1 / 共 2 个

undefined XP
    1
    2
  • 向模型中添加一个 dropout 层，以 50% 概率将元素置零。
  • 将模型设为 训练模式，使用 features 生成预测，并将输出保存为 output_train。