AstrAI/astrai/trainer
ViperEkura 84ed2327f5 feat: add --resume flag to decouple weight loading from training resumption
- Add --resume bool flag to train.py CLI
- --param_path always loads weights only by default
- --resume restores epoch, consumed_samples, optimizer & scheduler
- Checkpoint.load() now preserves full meta dict
- Update test_early_stopping to use new param_path/resume API
2026-07-16 14:23:23 +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 refactor: separate old policy and ref model in GRPO strategy 2026-07-14 20:03:45 +08:00
train_callback.py fix: init last_ckpt_step and last_log_flush_step from context.optimizer_step 2026-07-15 22:15:55 +08:00
train_context.py feat: add --resume flag to decouple weight loading from training resumption 2026-07-16 14:23:23 +08:00
trainer.py feat: add --resume flag to decouple weight loading from training resumption 2026-07-16 14:23:23 +08:00