Simple Convolution model does not work

1

Hi,

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

Thanks

2

Problem solved.

I forgot to do model.cuda()

Saeed