Hello, I have assembled a model from several small models.
I transferred the model to cuda and debugged to make sure it works with cuda,
on the other hand when I run the code I get the following error:
“Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same”
I would be happy if someone has an ideaת
Thanks,
Aviad