Donk10:

model(data_pixel2[i].unsqueeze(0))


I actually fixed that error, however I now face with this one :
RuntimeError Traceback (most recent call last)
 in () 1 for i in range(65536) : ----> 2 out_put[i] = model(data_pixel2[i].unsqueeze(0))
5 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py in forward(self, input) 200 _single(0), self.dilation, self.groups) 201 return F.conv1d(input, self.weight, self.bias, self.stride, --> 202 self.padding, self.dilation, self.groups) 203 204
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same