This website requires JavaScript.
Explore
Help
Register
Sign In
ViperEkura
/
AstrAI
Watch
1
Star
0
Fork
0
Code
Issues
Pull Requests
Packages
Projects
Releases
Wiki
Activity
Files
d7ac66fb73fc1aae255742f7e80ea89e5fc5dcaa
AstrAI
/
astrai
/
trainer
T
History
ViperEkura
d7ac66fb73
refactor: simplify attention mask handling
2026-07-20 20:36:16 +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
rollout.py
refactor: map instruction/input/output to chat roles
2026-07-20 13:55:25 +08:00
schedule.py
fix: scheduler division-by-zero, loss_mask bool
2026-07-03 22:04:55 +08:00
strategy.py
refactor: simplify attention mask handling
2026-07-20 20:36:16 +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
fix: inject LoRA before loading checkpoint state_dict
2026-07-20 17:00:24 +08:00
trainer.py
feat: add --resume flag to decouple weight loading from training resumption
2026-07-16 14:23:23 +08:00