Hi, I got the above error, and from the docs I’m not able to understand where should I adjust to cuda.
For example in the docs they are showing how to create a tensor on the gpu, but what with the case when I’m loading Images? still the same?
Now I have a situation my tensors are cuda type, but my layers not, and that’s why I’m getting :
"Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor ".
Now I need to put .cuda() on each layer?(or something like this) doesn’t make sense.
This is were the program stoped, in the nn.conv2d , tried to “.cuda()”
, didn’t help
             return nn.Sequential(
                nn.Conv2d(inp, oup, 3, stride, 1, bias=False).cuda(),
                nn.BatchNorm2d(oup),
                nn.ReLU(inplace=True)
            )

I tried few of the solutions online, but nothing worked.