neural network
justtoomuchforyou
没人在乎。
展开
-
整理pytorch报错
Stack Overflow[1]中有这个问题的描述,自己写了一个dice loss,没法反向传播,报这个错。原因应该是因为用了torch.argmax(),好像这个函数不可导,所以没法反向传播。但是...原创 2020-03-10 22:54:43 · 8904 阅读 · 8 评论 -
1*1卷积层的作用(pointwise convolution)
自己画了一张图:一个n*n,256个通道的feature map,经过一个1*1,64层的卷积核进行卷积,结果得到一个n*n,64通道的map,相当于发生了降维。即ResNet的设计的bottleneck的第一步:256-d经1x1,64进行降维。第二步从64通道到256通道同理。关于参数数量的减少具体参见这篇博客:https://blog.csdn.net/renhaofan...原创 2019-09-04 14:31:41 · 4010 阅读 · 2 评论