My current output is [num_classes], but why isn’t throwing an error before that? Also, it’s NLLoss taking care of making it into a one-hot vector?
My current output is [num_classes], but why isn’t throwing an error before that? Also, it’s NLLoss taking care of making it into a one-hot vector?