尝试使用 dropout
Dropout 通过在训练过程中随机将部分输出值置零来帮助防止过拟合。在本练习中，您将构建一个带有 dropout 的简单神经网络，并观察它在训练模式与评估模式下的表现。
torch.nn 包已作为 nn 预加载，features 也已为您定义。
本练习是课程的一部分
使用 PyTorch 的深度学习入门
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Model with Dropout
model = nn.Sequential(
nn.Linear(8, 6),
nn.Linear(6, 4),
____)
# Forward pass in training mode (Dropout active)
model.____
output_train = ____