transformers
refactor _inner_training_loop to smaller methods
#44041
Merged

refactor _inner_training_loop to smaller methods #44041

winglian
winglian refactor _inner_training_loop to smaller methods
a50e35e2
HuggingFaceDocBuilderDev
SunMarc
SunMarc commented on 2026-02-16
winglian address PR comments and remove unused args
4bbbac19
SunMarc move _train_batch_size
48391e9e
SunMarc big update
cda1b847
SunMarc Merge remote-tracking branch 'origin/main' into refactor-inner-traini…
f52ffbf4
SunMarc
SunMarc move DummyScheduler
c9f36103
SunMarc style
f4cad42b
SunMarc make it private
46e20417
SunMarc switch to kwargs_handlers
da07a717
SunMarc style
c6bee6a7
SunMarc fix resuming
daec1155
SunMarc style
a4347362
SunMarc revert loading after
fa551548
SunMarc way better now
c18c1677
SunMarc move method
2ced96a5
SunMarc small mistake
7feecabe
SunMarc update
b37736d9
winglian winglian force pushed from 63963355 to b37736d9 37 days ago
SunMarc SunMarc requested a review from kashif kashif 37 days ago
SunMarc SunMarc requested a review from qgallouedec qgallouedec 37 days ago
SunMarc
SunMarc
SunMarc remove tp_size
a697854f
SunMarc
SunMarc account for grpo
cc244320
qgallouedec
qgallouedec
qgallouedec commented on 2026-02-20
qgallouedec
qgallouedec
winglian
winglian commented on 2026-02-21
SunMarc final touch
2e97639a
SunMarc
SunMarc approved these changes on 2026-02-23
SunMarc
SunMarc Merge branch 'main' into refactor-inner-training-loop-reorder-only
23a6e8bf
winglian winglian merged f26814d9 into main 34 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone