When I run my model, it returns me Input type (CUDAFloatTensor) and weight type (CPUFloatTensor) should be the same
I think that’s because I used a pretrained model and used the weights of that pretrained model maybe, because the error returns from there.
Any idea to fix it?