【Pytorch实用教程】循环神经网络中使用dropout需要注意的问题
文章目录
-
- 问题
- 解答
-
- 警告的具体含义
- 解决方案
- 示例代码
- 总结
问题
UserWarning: dropout option adds dropout after all but last recurrent layer, so non-zero dropout expects num_layers greater than 1, but got dropout=0.3 and num_layers=1
warnings.warn("dropout option adds dropout after all but last "
解答
这个警告来自 PyTorch 的循环神经网络（如 nn.LSTM 或 nn.GRU