This website requires JavaScript.
Explore
Help
Register
Sign In
ViperEkura
/
AstrAI
Watch
1
Star
0
Fork
You've already forked AstrAI
0
Code
Issues
Pull Requests
Packages
Projects
Releases
Wiki
Activity
6269bacfc3
AstrAI
/
astrai
History
ViperEkura
6269bacfc3
refactor: decode 按页分桶批处理,position_ids 改为 per-task 构建
2026-05-14 14:22:11 +08:00
..
config
refactor: 移除 device_ids 参数设计,统一通过 CUDA_VISIBLE_DEVICES 控制 GPU 分配;更新 README 训练示例
2026-05-09 14:55:43 +08:00
dataset
fix: MultiSegmentFetcher 空 dict 崩溃 + BaseDataset assert 替换为显式 raise
2026-05-12 11:41:45 +08:00
inference
refactor: decode 按页分桶批处理,position_ids 改为 per-task 构建
2026-05-14 14:22:11 +08:00
model
refactor: 位置编码改用 position_ids [B,S],简化 attention mask 构建
2026-05-14 13:26:31 +08:00
parallel
refactor: 移除 device_ids 参数设计,统一通过 CUDA_VISIBLE_DEVICES 控制 GPU 分配;更新 README 训练示例
2026-05-09 14:55:43 +08:00
tokenize
fix: 修复 CLI 参数缺失/重复、device_ids 越界、generate 参数名不一致、scheduler 时序、非流式截断等 bug
2026-05-09 14:36:42 +08:00
trainer
feat: Checkpoint 支持 extra 通用扩展数据,用户通过函数自定义保存/恢复优化器等状态
2026-05-09 15:50:38 +08:00
__init__.py
release: v1.3.4
2026-05-10 15:59:18 +08:00
factory.py
chore: 修改类名,优化导入顺序
2026-04-05 22:27:57 +08:00
serialization.py
refactor: 抽取 BaseStorage 存储抽象,支持 JSON 原始文本数据加载
2026-05-12 11:17:24 +08:00