NonLocalandSEnet

NonLocalandSEnet copied to clipboard

I use the mx.sym.batch_dot to calculate matrix multiplication, but my network report error: ValueError: Input contains NaN, infinity or a value too large for dtype('float64'). why???

Aruen24

SE , SGE

3

作者你好，我想改动一个新的结构，是在SE的地方改动的，有点困惑，mxnet 的symbol，不能直接得到bchw的值， pytorch 的SGE，一个实现架构语句， 对应你提供的模型SE代码位置修改的话，symbol每一层bn3 后边的bchw,我直接得不到，我要mxnet,实现这句话，b, c, h, w = x.size()， x = x.reshape(b * self.groups, -1, h, w) 我对mxnet 不是那么熟悉，不知道作者你有没有好的方式实现这句reshape 我在修改的地方 bn3 = mx.sym.BatchNorm(data=conv2, fix_gamma=False, eps=2e-5, momentum=bn_mom, name=name +...

shiyuanyin