transformers
ebdd1ad4 - Pass correct `num_items_in_batch` value into the training_step function (#35438)

Commit

1 year ago

Pass correct `num_items_in_batch` value into the training_step function (#35438) pass correct `num_items_in_batch` to compute_loss

References

Author

hiyouga

Parents