Pass correct `num_items_in_batch` value into the training_step function #35438
pass correct `num_items_in_batch` to compute_loss
4e7b4c37
hiyouga
changed the title Pass correct `num_items_in_batch` value into the training_step method Pass correct `num_items_in_batch` value into the training_step function 1 year ago
Merge branch 'main' into patch-11
36ecc23f
muellerzr
approved these changes
on 2025-01-07
SunMarc
approved these changes
on 2025-01-07
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub