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
b558e61f6390a5e6da33288b8c5bd0f87514a8e5
AstrAI
/
astrai
/
trainer
T
History
ViperEkura
b558e61f63
refactor: 简化 _disable_random_init,scheduler 移入同步块
...
- _disable_random_init: enable=False 提前返回,dict 推导替代空字典 - scheduler.step() 移入 sync_gradients 守卫内
2026-05-26 17:05:25 +08:00
..
__init__.py
feat: 新增 Muon 优化器
2026-05-17 16:44:03 +08:00
metric_util.py
feat: 训练中新增验证循环
2026-05-17 16:12:42 +08:00
optim.py
perf: Muon step 改用 torch._foreach_* 批处理并移除 NS 迭代的冗余 bf16 转换
2026-05-23 19:50:12 +08:00
schedule.py
feat: 优化工厂模式的实现
2026-04-04 15:49:46 +08:00
strategy.py
feat: 新增FSDP并行后端
2026-05-25 19:43:14 +08:00
train_callback.py
fix: ProgressBar默认输出到stdout
2026-05-26 13:27:05 +08:00
train_context.py
fix: 断点续训恢复优化器/调度器状态及采样器剩余长度
2026-05-26 13:50:25 +08:00
trainer.py
refactor: 简化 _disable_random_init,scheduler 移入同步块
2026-05-26 17:05:25 +08:00