Simple Convolution model does not work


I’m building a very simple model with only one conv-BatchNorm and Relu. But when I feed much input x, the following error occurs. Any one knows the reason?

RuntimeError: Expected object of type CUDAFloatType but found type CPUFloatType for argument #3 ‘weight’

My x is already CUDAFloatType


Problem solved.

I forgot to do model.cuda()