AstrAI

Commit Graph

Author	SHA1	Message	Date
ViperEkura	a548d4553e	fix: 断点续训恢复优化器/调度器状态及采样器剩余长度 - 使用Checkpoint.load()替代手动加载model.safetensors，恢复optimizer/scheduler状态 - TrainContextBuilder从checkpoint.extra恢复优化器和调度器state_dict - ResumableDistributedSampler.__len__返回剩余样本数而非总数 - 训练前对state_dict置空避免mp.spawn pickle 7GB大对象	2026-05-26 13:50:25 +08:00
ViperEkura	9c31d78a22	chore: 将data 模块命名为dataset	2026-04-04 16:16:27 +08:00

Author

SHA1

Message

Date

ViperEkura

a548d4553e

fix: 断点续训恢复优化器/调度器状态及采样器剩余长度

- 使用Checkpoint.load()替代手动加载model.safetensors，恢复optimizer/scheduler状态
- TrainContextBuilder从checkpoint.extra恢复优化器和调度器state_dict
- ResumableDistributedSampler.__len__返回剩余样本数而非总数
- 训练前对state_dict置空避免mp.spawn pickle 7GB大对象

2026-05-26 13:50:25 +08:00

ViperEkura

9c31d78a22

chore: 将data 模块命名为dataset

2026-04-04 16:16:27 +08:00

2 Commits