transformers

Pass num_items_in_batch directly to loss computation

#36753

Merged