博主在模型推理过程中遇到模型结果不一致的问题，发现是由于F.dropout2d在推理时未将p设为0。在评估模式(model.eval())下，nn.Dropout会自动处理dropout，避免了每次推理结果的变化。理解并正确使用dropout在模型管理和推理中至关重要。
今天在模型推理时，对模型设置为model.eval()之后，每次结果都不一样，感到很奇怪，经过调试找到原因，采用F.dropout2d时（下载的别人的模型），在推理时，需要手动将p设置为0，否则就每次不一样了，为了便于模型管理采用nn.Dropout，因为eval()，会自动处理dropout的。
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
