Similar problem occured saying
RuntimeError: log_softmax_forward is not implemented for type torch.LongTensor
When using nn.CrossEntropyLoss() ( But works in MSELoss)
Before i was getting
RuntimeError: Expected object of type torch.LongTensor but found type torch.FloatTensor for argument #2 'target’
I’ve changed the tensor type to Longtensor by using .long() for both input and label images in criterion as
        loss = criterion(outputs.long(), images.long())
Now getting the this error.I’ve changed both to other type and different type each giving different errors.
Please help!!.