I have 2 numpy arrays, which I convert into tensors to use the TensorDataset object.
import torch.utils.data as data_utils
X = np.zeros((100,30))
Y = np.zeros
This is because in PyTorch, you can not do operations between Tensor of different types. Your data
is DoubleTensor
, but the model parameter are FloatTensor
. So you get this error message. As @mexmex have said, convert data
to FloatTensor
to make it conform with the model parameter type.
Do not do the other way around! Trying to convert the model to double is greatly discouraged by PyTorch devs as GPUs are not good at double precision computation. Also, floating point is pretty enough for deep learning.