The .fit() method passes unsized batches to a custom loss function.

The .fit() method passes unsized batches to a custom loss function. #969