Hi,
my inputs into the loss functions are shaped as NCHW (8, 2, 256, 256) for y_pred and as NHW (8, 256, 256) for y_true. Here C mean number of classes. During the training phase, nn.CrossEntropyLoss() works well but other loss functions like L1Loss receives errors due to that number of classes. Please advise!