The .fit() method passes unsized batches to a custom loss function. #19732
Labels
type:support
User is asking for help / asking an implementation question. Stackoverflow would be better suited.
The .fit() method passes unsized batches to a custom loss function if the dataset size is not a multiple of the batch size.
This happens twice, then everything goes fine. But if you need to use the batch size to normalize or reshape data, then an error occurs.
The text was updated successfully, but these errors were encountered: