got this error :-
Epoch  0  /  90

RuntimeError                              Traceback (most recent call last)
 in ()
1 if name == “main”:
2
---->3   test(data_loader)
4
4 frames
 in forward(self, x)
11         weight = weight / std.expand_as(weight)
12         return F.conv2d(x, weight, self.bias, self.stride,
—> 13                         self.padding)
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
while using this


image1188×885 54.7 KB