Wei

@fmassa Yes, I did this. When I use cudnn.convert(model, cudnn, function(module) return torch.type(module):find('SpatialBatchNormalization') end), it works.

## Maybe nn.Identity()() caused the problem. ``` lua require 'nngraph' require 'cutorch' require 'cunn' require 'cudnn' local input = nn.Identity()() local features = nn.Sequential() features:add(nn.SpatialConvolution(1,8,5,5,1,1,2,2)) features:add(nn.Abs()) features:add(nn.SpatialBatchNormalization(8,nil,nil,false)) features:add(nn.Tanh()) features:add(nn.SpatialAveragePooling(5,5,2,2,2,2)) features:add(nn.SpatialConvolution(8,1,5,5,1,1,2,2))...

Me too