AstrAI/astrai
ViperEkura 6715461a36 chore : 升级 torch 2.11.0+cu128,移除自定义 Muon,修复 gloo device_id
- torch 2.7.1-cu126 升级至 2.11.0-cu128,numpy 2.3.2 升级至 2.4.4
- 移除 astrai/trainer/optim.py,改用 torch.optim.Muon
- parallel setup: gloo 后端不再传递 device_id,单卡多进程不再报错
2026-06-27 16:10:37 +08:00
..
config refactor : neftune_alpha 在 Embedding 构造时传入,由模型配置链路负责 2026-06-19 14:23:27 +08:00
dataset refactor : 清理工厂和配置系统中的死代码与冗余抽象 2026-06-07 11:39:50 +08:00
inference fix: 使用 threading.Event 替代裸 bool,补全公共 API 2026-06-18 15:38:35 +08:00
model feat : GPT-2 residual scaling weight init 2026-06-25 15:08:31 +08:00
parallel chore : 升级 torch 2.11.0+cu128,移除自定义 Muon,修复 gloo device_id 2026-06-27 16:10:37 +08:00
preprocessing fix: 修复预处理流水线 4 个致命问题 2026-06-18 17:38:01 +08:00
tokenize refactor : Pipeline 去除去重,ids 重命名为 sequence,泛型透传 2026-05-31 15:14:27 +08:00
trainer chore : 升级 torch 2.11.0+cu128,移除自定义 Muon,修复 gloo device_id 2026-06-27 16:10:37 +08:00
__init__.py chore : 升级 torch 2.11.0+cu128,移除自定义 Muon,修复 gloo device_id 2026-06-27 16:10:37 +08:00
factory.py refactor : metric_logger 改用事件类型 (type=step/validation/epoch) 2026-06-25 17:18:20 +08:00
protocols.py refactor: 重构训练后端为 Executor 模式 2026-05-24 20:35:44 +08:00
serialization.py fix : resume_dir 无权重文件时不强制加载,支持仅配置训练 2026-06-13 15:40:14 +08:00