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