在写CBAM时遇到问题,maxpooling和average pooling要共享一个全连接,但是不会写这种共享一个层的。
写法:
先定义出层不用加输入,相当于一个函数
然后用的时候直接调用就好了。
感谢大神!!参考:https://blog.csdn.net/ngadminq/article/details/107005092
在写CBAM时遇到问题,maxpooling和average pooling要共享一个全连接,但是不会写这种共享一个层的。
写法:
先定义出层不用加输入,相当于一个函数
然后用的时候直接调用就好了。
感谢大神!!参考:https://blog.csdn.net/ngadminq/article/details/107005092
相关文章: