AstrAI

Commit Graph

Author	SHA1	Message	Date
ViperEkura	ca4e6b907c	feat: Checkpoint 支持 extra 通用扩展数据，用户通过函数自定义保存/恢复优化器等状态 - serialization.py: Checkpoint 新增 extra: dict 字段， save() 写入 extra.pt，load() 自动恢复 - train_callback.py: CheckpointCallback 新增 save_extra_fn 参数，用户传入 (context) -> dict 决定保存哪些额外状态 - train_context.py: TrainContextBuilder 新增 load_extra_fn 参数，用户传入 (extra, context) 从 checkpoint 恢复状态	2026-05-09 15:50:38 +08:00
ViperEkura	bd9741dc5f	refactor: 从data 模块分离tokenizer	2026-04-04 16:12:58 +08:00

Author

SHA1

Message

Date

ViperEkura

ca4e6b907c

feat: Checkpoint 支持 extra 通用扩展数据，用户通过函数自定义保存/恢复优化器等状态

- serialization.py: Checkpoint 新增 extra: dict 字段，
  save() 写入 extra.pt，load() 自动恢复
- train_callback.py: CheckpointCallback 新增 save_extra_fn
  参数，用户传入 (context) -> dict 决定保存哪些额外状态
- train_context.py: TrainContextBuilder 新增 load_extra_fn
  参数，用户传入 (extra, context) 从 checkpoint 恢复状态

2026-05-09 15:50:38 +08:00

ViperEkura

bd9741dc5f

refactor: 从data 模块分离tokenizer

2026-04-04 16:12:58 +08:00

2 Commits