SpatialConvolution : assertion failed!

[Ahmed Osama]

I am using the following command to train the model

th train.lua -data dataset -style_image ./AdobeStock_90768625.jpeg -style_size 600 -image_size 512 -model johnson -batch_size 4 -learning_rate 1e-2 -style_weight 10 -style_layers relu1_2,relu2_2,relu3_2,relu4_2 -content_layers relu4_2 -cpu

but this error is raised :-

/home/src/torch/install/bin/luajit: ...erspace/src/torch/install/share/lua/5.1/nn/Container.lua:67: 
In 1 module of nn.Sequential:
...aperspace/src/torch/install/share/lua/5.1/cudnn/init.lua:171: assertion failed!
stack traceback:
	[C]: in function 'assert'
	...aperspace/src/torch/install/share/lua/5.1/cudnn/init.lua:171: in function 'toDescriptor'
	...torch/install/share/lua/5.1/cudnn/SpatialConvolution.lua:123: in function 'createIODescriptors'
	...torch/install/share/lua/5.1/cudnn/SpatialConvolution.lua:188: in function <...torch/install/share/lua/5.1/cudnn/SpatialConvolution.lua:186>
	[C]: in function 'xpcall'
	...erspace/src/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
	...rspace/src/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'
	./src/descriptor_net.lua:28: in function 'forward'
	train.lua:212: in function 'opfunc'
	...aperspace/src/torch/install/share/lua/5.1/optim/adam.lua:37: in function 'optim_method'
	train.lua:241: in main chunk
	[C]: in function 'dofile'
	.../src/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk
	[C]: at 0x00406670

WARNING: If you see a stack trace below, it doesn't point to the place where this error occurred. Please use only the one above.
stack traceback:
	[C]: in function 'error'
	...erspace/src/torch/install/share/lua/5.1/nn/Container.lua:67: in function 'rethrowErrors'
	...rspace/src/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'
	./src/descriptor_net.lua:28: in function 'forward'
	train.lua:212: in function 'opfunc'
	...aperspace/src/torch/install/share/lua/5.1/optim/adam.lua:37: in function 'optim_method'
	train.lua:241: in main chunk
	[C]: in function 'dofile'
	.../src/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk
	[C]: at 0x00406670