Open Rechardgu opened 6 years ago
RuntimeError: softmax(): argument 'input' (position 1) must be Variable, not torch.FloatTensor
It depends on the version of torch you are using (it's fixed with the last version 0.4):
You should add this to hack it:
softmax(torch.autograd.Variable(input, requires_grad=False))
RuntimeError: softmax(): argument 'input' (position 1) must be Variable, not torch.FloatTensor