Lets say we are defining the loss function of a neural network as follows:
classifier_loss = classifier_criterion(classifier_output, label)
Is it true that classifier_output needs to be a FloatTensor and label as a LongTensor ??
I have batch size =1 thats why my label is a 1x1 ByteTensor. Do I need to convert into a LongTensor ??..
If yes, how do I do that ??
I am using crossentropyloss.
Thanks in advance