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