torch7 error

欢迎使用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)