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
5e73ca20aa8165e4b7ccd52db3fccee344349444
AstrAI
/
astrai
/
trainer
T
History
ViperEkura
9fe2121743
feat : TrainConfig 支持 val_split 从训练集自动切分验证集
...
- val_split 比例从 dataset 中划出验证集,用 random_seed 固定随机切分 - 若 val_dataset 已显式设置则跳过自动切分
2026-06-02 20:33:40 +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
refactor : 移除 -> None 返回值标注,拆分 FSDP 参数,新增 mmap 数据集存储
2026-05-28 13:57:06 +08:00
strategy.py
fix : FSDP 优化器顺序、温度除零、调度器静默死亡、ref模型设备
2026-05-29 21:57:44 +08:00
train_callback.py
fix : 并行训练 state_dict 收集与训练/推理并发缺陷
2026-05-29 21:12:52 +08:00
train_context.py
feat : TrainConfig 支持 val_split 从训练集自动切分验证集
2026-06-02 20:33:40 +08:00
trainer.py
refactor: 检查点加载重构,路径替代对象传递
2026-05-27 20:15:29 +08:00