feat : checkpoint 支持保存 config.json

- Checkpoint.save 写入独立的 config.json(模型架构参数)
- Checkpoint.load 读取 config.json,恢复时覆盖 context.model_config
- TrainContext 新增 model_config 字段,builder 从 resume_dir/config.json 加载
- BaseConfig.to_dict 支持 tuple 和嵌套 dataclass(如 LoRAConfig)
- 删除 _get_meta/_get_config wrapper,直接使用 load_json
This commit is contained in:
2026-05-28 20:21:51 +08:00
parent 3a28e52e98
commit c424dfc293
4 changed files with 33 additions and 17 deletions
+1 -1
View File
@@ -160,7 +160,7 @@ class CheckpointCallback(TrainCallback):
epoch=context.epoch,
iteration=context.iteration,
extra=extra,
meta=context.config.to_dict(),
config=context.model_config,
)
context.checkpoint.save(save_path)