Problem creating custom loss function

fastai
2

Try: learn.loss_func = MyCrossEntropyFlat(axis=1), thats the channel that indicates the labels.

3 Likes