torch.nn.dropout和torch.nn.F.dropout区别

  • Dropout被设计为只在训练中使用,所以当你对模型进行预测或评估时,你需要关闭Dropout。nn.dropout可以方便地处理这个问题,在模型进入eval时立即关闭Dropout,而F.dropout并care你是什么模式。

  • 分配给模型的所有模块都在模型中注册。所以模型类跟踪它们,这就是为什么可以通过调用eval()关闭Dropout模块。当使用F.dropout时,你的模型并不知道它,所以模型的summary中也不会出现dropout模块。

https://blog.csdn.net/r1254/article/details/91867736

猜你喜欢

转载自blog.csdn.net/qq_42363032/article/details/126747440