在网络中加入SE模块后报错:AttributeError: 'NoneType' object has no attribute 'fill_'

原因:SE模块的全连接偏置设置为false,而权重初始化时置零。

解决方法:将初始化偏置注释掉

或者将全连接中的偏置设置为ture 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐