浅谈pytorch池化maxpool2D注意事项
更新时间:2020年02月18日 10:31:15 作者:陈军2号
今天小编就为大家分享一篇浅谈pytorch池化maxpool2D注意事项,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
注意: 在搭建网络的时候用carpool2D的时候,让高度和宽度方向不同池化时,
用如下:
nn.MaxPool2d(kernel_size=2, stride=(2, 1), padding=(0, 1))
千万不要用:
nn.MaxPool2d(kernel_size=2, stride=(2, 1), padding=(0, 0)), 这样在用交叉熵做损失函数的时候,有时候会出现loss为nan的情况,检查的时候发现,某些样本的提取出来的feature全为nan。
以上这篇浅谈pytorch池化maxpool2D注意事项就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Pandas 合并多个Dataframe(merge,concat)的方法
今天小编就为大家分享一篇Pandas 合并多个Dataframe(merge,concat)的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2018-06-06
Python使用textract实现从各种文件中提取文本信息
textract是一个强大的Python库,可以用于从各种文件格式中提取文本,本文将介绍textract的使用场景,以及一些常用的Python代码案例,希望对大家有所帮助2024-01-01


最新评论