Hi all,
I’m facing a cryptic error which I can’t find any discussion about.
I’m simply trying to pass the output of a LSTM layer to a linear layer. The code is as follows:
output, hidden = hidden
logits = self.fully_connected(output)
fully_conected is simply nn.Linear. and output is a torch.cuda.FloatTensor of size (batch_size, n_hidden).
The error message I’m receiving is the following:
TypeError: CudaThreshold_updateOutput received an invalid combination of arguments - got (int, torch.cuda.FloatTensor, torch.cuda.FloatTensor, int, int, Linear), but expected (int state, torch.cuda.FloatTensor input, torch.cuda.FloatTensor output, float threshold, float val, bool inplace)
Does anyone know what might be the cause of this error?