AI模型复用关键在于完整保存结构、参数和预处理逻辑，并严格匹配加载流程；需用checkpoint保存权重与配置，单独序列化预处理对象，记录超参等信息，加载时先重建结构再加载权重并验证输出。
训练好的AI模型要复用，关键不是“重新跑一遍”，而是把训练结果**完整、准确、可迁移地保存下来，再在需要时原样加载运行**。核心在于保存什么、怎么保存、加载后如何正确使用——尤其要注意模型结构、参数、预处理逻辑三者必须匹配。
保存模型：结构+权重+配置缺一不可
只保存模型权重（如 model.state_dict()）最轻量，但加载时必须先用完全一致的代码重建模型结构；只保存整个模型对象（如 torch.save(model, ...)）看似方便，却可能因PyTorch版本或自定义类路径变化而失败。更稳妥的做法是：
- 用 torch.save({'model_state_dict': model.state_dict(), 'optimizer_state_dict': optimizer.state_dict(), 'epoch': epoch, 'loss': loss}, path) 保存检查点（checkpoint），兼顾复现训练和推理
- 同时单独保存输入预处理逻辑（如Tokenizer、StandardScaler对象），用 joblib 或 pickle 存为 .pkl 文件
- 记录关键配置：模型超参、输入尺寸、类别名列表（class_names）、归一化均值/标准差等，写入 JSON 或直接注释在保存脚本中
加载模型：先搭结构，再灌参数，最后验证输出
加载不是“一行代码完事”。典型安全流程是：
- 用原始代码（或等效定义）**重新实例化模型类**，确保结构完全一致
- 调用 model.load_state_dict(torch.load(path)['model_state_dict']) 加载权重
- 立即执行 model.eval()（关闭 dropout/batch norm 更新）
- 用一个已知输入（如全零张量或固定样本）前向推理，检查输出 shape 和 dtype 是否符合预期 —— 这步能快速暴露结构错位或设备不匹配问题
跨环境复用：避开路径、设备、版本三大坑
模型在A机器训练，到B机器加载失败？常见原因有：
立即学习“Python免费学习笔记（深入）”；
- 设备不一致：保存时在GPU上，加载时没指定 map_location。正确写法：torch.load(path, map_location='cpu')（部署时优先加载到CPU）
- 路径硬编码：保存/加载路径写死在代码里。建议用相对路径 + os.path.join(os.path.dirname(__file__), 'models', 'best.pt')
- 版本漂移：PyTorch 2.0 保存的模型，在 1.12 上可能无法读取。生产环境应锁定框架版本，并在 README 中注明
轻量部署：转 ONNX 或 TorchScript 更适合落地
如果目标是嵌入式、Web 或非Python环境，原生 PyTorch 模型不够友好。推荐两步走：
- 导出为 ONNX：torch.onnx.export(model, dummy_input, 'model.onnx', opset_version=12)，支持 C++、Java、JS 多平台推理
- 或编译为 TorchScript：scripted = torch.jit.script(model); scripted.save('model.pt')，保留 Python 语义且无需解释器，启动更快
- 无论哪种，导出前务必用相同输入做一次前向验证，确认输出与原始模型一致
模型复用不是技术炫技，而是让训练成果真正变成可交付、可维护、可迭代的资产。保存时多记一行配置，加载时多验一次输出，省下的调试时间远超初期那几分钟。
