How to set the type of tensor

1

When I write my dataloader, I found the data were transform into DoubleTensor, However when I load some pretrained model, I found my model weights were FloatTensor, How to solve this problem

2

Maybe modifying your dataloader to transform the data into FloatTensor would be a way to solve your problem ?

3

Or this simple hack…

output = model(input.float())
2 Likes
4

yes, I did that, I make all the tensor into floatTensor