Hi,
I have a model with output shape: (batch_size, N, M) and target shape: (batch_size, N, M)
this is a classifier.
so I should use NLL or CrossEntropy as the loss function.
but torch gives me 1D target tensor expected, multi-target not supported error
how can I fix this?