Dropout
1 | def dropout(X, drop_prob): |
对于用torch封装好的dropout,只需在定义网络结构时在需要dropout的层后添加 Dropout层即可,在测试模型(model.eval())时,它不会触发。
1 | net = nn.Sequential( |
1 | def dropout(X, drop_prob): |
对于用torch封装好的dropout,只需在定义网络结构时在需要dropout的层后添加 Dropout层即可,在测试模型(model.eval())时,它不会触发。
1 | net = nn.Sequential( |