🍌 前言
# 前言
本文档是对[代码部署规范](https://conflu.hdec.com/pages/viewpage.action?pageId=39356626)的拓展。
# 算法开发文件结构

部署文件和训练文件都需要上传,两者需要使用文件夹进行隔离。
>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")
```