欢迎使用Markdown编辑器写博客
net:add(nn.SpatialBatchNormalization())
用法及错误处理:
例:
net:add(nn.SpatialConvolution(1,64,3,3))
net:add(nn.SpatialBatchNormalization(64))
net:add(nn.ReLU())
net:add(nn.SpatialMaxPooling(2,2,2,2))
错误提示torch/install/share/lua/5.1/nn/BatchNormalization.lua:78: only mini-batch supported (4D tensor), got 3D tensor instead
解决方法:该错误是对输入的要求为四维向量,而输入向量为三维向量,将输入调整为四维向量就可以了:
input = input:reshape(1,1,28,28)