ViperEkura b092316385 feat : add distributed checkpoint via executor checkpoint_context ...

- Add checkpoint_context context manager to BaseExecutor with entry/exit barrier
- Add _gather_state_dict hook overridden per executor (template method)
- DDPExecutor skips unwrap on non-rank-0 to avoid redundant state_dict gather
- FSDPExecutor uses rank0_only=True to reduce memory on non-writers
- Remove redundant rank-0 guard from Checkpoint.save and manual barrier from Callback

2026-07-13 12:27:09 +08:00

..

__init__.py

chore : 升级 torch 2.11.0+cu128，移除自定义 Muon，修复 gloo device_id

2026-06-27 16:10:37 +08:00

metric_util.py

refactor : grad_norm 指标简化，clip_grad_norm 移至 executor

2026-06-30 14:59:43 +08:00

schedule.py

fix: scheduler division-by-zero, loss_mask bool

2026-07-03 22:04:55 +08:00

strategy.py

fix: token-level ratio and prompt masking in GRPO strategy

2026-07-12 21:24:09 +08:00

train_callback.py

feat : add distributed checkpoint via executor checkpoint_context

2026-07-13 12:27:09 +08:00

train_context.py

refactor : replace iteration with consumed_samples

2026-06-30 18:42:42 +08:00

trainer.py

refactor: merge validation into MetricCallback, simplify progress bar to optimizer steps

2026-07-03 21:43:08 +08:00