Predict - Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

[dorienh]

Predict function still has the loss function error I believe.

----> 3 model.predict(testdata=process.X_test, target=process.y_test)
      4 
      5 

7 frames
/usr/local/lib/python3.7/dist-packages/torch/nn/modules/conv.py in _conv_forward(self, input, weight, bias)
    258                             _single(0), self.dilation, self.groups)
    259         return F.conv1d(input, weight, bias, self.stride,
--> 260                         self.padding, self.dilation, self.groups)
    261 
    262     def forward(self, input: Tensor) -> Tensor:

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

[Luckygyana]

9