更新训练脚本和推理工具，调整数据集结构和模型加载逻辑

更新训练脚本和推理工具，调整数据集结构和模型加载逻辑 #10