🍌 前言

# 前言 本文档是对[代码部署规范](https://conflu.hdec.com/pages/viewpage.action?pageId=39356626)的拓展。 # 算法开发文件结构 ![项目_.png](https://cos.easydoc.net/12310430/files/lepg5211.png) 部署文件和训练文件都需要上传,两者需要使用文件夹进行隔离。 >i 常见的做法是将部署文件放入名叫Docker的文件夹 # 模型保存 和pytorch框架相关的模型大致分为只保存参数的picked模型和可序列化的script模型。 只保存模型参数方式: ```python module = MyModule() torch.save(module.state_dict(), "picked_model.pth") ``` TorchScript模型保存方式: ``` module = MyModule() m = torch.jit.script(module) torch.jit.save(m, 'script_model.pt') ##### 等价于 ##### m.save("script_model.pt") ```