Creating Dataloaders for classification task with tabular

Turns out, fastai library has loss-function that digests tensors of any shapes by flattening them:

They’re called MSELossFlat, CrossEntropyLossFlat etc

1 Like