transformers
a75b9ffb - Update Loss Functions to Accept Tensor num_items_in_batch (#38029)

Commit
278 days ago
Update Loss Functions to Accept Tensor num_items_in_batch (#38029) * Update Loss Functions to Accept Tensor num_items_in_batch * Fix device mismatch by moving num_items_in_batch to loss device in fixed_cross_entropy * fix the ruff check * delete the unused if stat * fix the type problem
Author
Parents
Loading