transformers
Update Loss Functions to Accept Tensor num_items_in_batch
#38029

Merged

Update Loss Functions to Accept Tensor num_items_in_batch #38029

ArthurZucker merged 7 commits into huggingface:main from iamsernine:main

Update Loss Functions to Accept Tensor num_items_in_batch

ada5e324

github-actions marked this pull request as draft 304 days ago

iamsernine marked this pull request as ready for review 304 days ago

github-actions requested a review from

ArthurZucker 304 days ago

SunMarc dismissed these changes on 2025-05-12

SunMarc dismissed their stale review 301 days ago

missclick

Fix device mismatch by moving num_items_in_batch to loss device in fi…

2b5c012f

iamsernine requested a review from

SunMarc 294 days ago

fix the ruff check

0825472d

delete the unused if stat

21783176

Merge branch 'main' into main

a4946e7b

fix the type problem

e3d1ed2f

Merge branch 'main' of github.com:NEREUScode/transformers

6eb9939a

iamsernine commented on 2025-05-19

ArthurZucker approved these changes on 2025-06-02

ArthurZucker merged a75b9ffb into main 280 days ago

Reviewers

ArthurZucker

SunMarc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone