保存Notebook中的模型
更新时间:2023-01-18
保存Notebook中的模型
模型只有保存为特定版本后,才可以执行检验,校验通过后可以将其发布到模型仓库。
请确保要保存的模型文件在/home/work/PretrainedModel目录下。模型支持版本管理功能,在保存时可以生成新版本也可以覆盖已有的且尚未部署的模型版本,每个版本的模型都可以独立部署。每个模型版本中保存的模型文件大小上限为1.5GB。
在保存模式时也可以将训练模型的代码一并保存。代码支持版本管理功能,用户再次启动Notebook时,可以使用指定的代码版本来初始化Notebook工作空间即/home/work目录下data以外的空间。每个代码版本中保存的文件大小上限为150M。
- 单击Notebook页面左侧工具栏中的“生成模型版本”。
-
在“保存模型”对话框中配置模型属性,如下所示:
- 生成新版本:生成一个新的模型版本,版本号由系统自动生成,若当前最高版本为Vn,则新生成版本为Vn+1。注意需要保存模型网络结构文件、参数文件以及标签文件(如:label.txt)。
- AI框架:即生成模型时使用的Paddle版本。
- 代码版本:即生成模型时使用的代码版本。
-
单击“下一步”,选择要保存的模型文件:
当前项目文件为PreitraindModel目录下的文件列表。
-
单击“生成”,完成模型保存。
在完成模型保存后,可以关闭Notebook,从而避免产生不必要的费用。