使用Notebook开发模型
所有文档

          BML 全功能AI开发平台

          使用Notebook开发模型

          使用Notebook开发模型

          在Notebook可以导入已创建的数据集并将其用于模型训练。

          预置代码说明

          当您创建的是图像分类、物体检测类型的Notbeook时,系统会自动生成一份代码,您只需要导入数据集并将代码做简单的适配即可快速创建。

          当您创建的是通用型的Notebook时,系统会根据您使用的框架来生成一份示例代码。

          导入数据集

          通过导入操作可以将已创建的数据集下载到Notebook的运行环境中,导入的数据集会保存在“data”目录下以数据集ID命名的文件夹中,对于已标注的图片,下载内容包括原始图片文件和同名的json标注文件。

          1. 单击Notebook左侧页签中的“数据集”。
          2. 在操作面板中,单击“导入数据集”,系统在弹出框中显示已创建的数据集及其标注情况,如下所示:

            fig05.png

          3. 勾选待导入数据集前的复选框并单击“确认”,开始导入数据集,如下所示:

            fig06.png

            数据集导入操作是单向的,且当数据服务中的数据集有更新时,不会自动同步,需要手动重新刷新。

          训练模型

          用户可以在Notebook或终端中编写模型训练代码,并完成模型的训练。

          本节给出如何使用示例代码完成导入数据集的建模。示例代码结构如下所示:

          fig002.png

          用户只需要执行train.sh脚本即可启动模型训练,其输入参数如下所示:

          train.sh <数据目录> <模型保存目录>

          • 数据目录:数据集保存目录的绝对路径
          • 模型保存目录:请设置为/home/work/PretrainedModel/,否则会造成生成新模型版本时无法读取模型文件。

          上述代码结构不是必须的,用户可以自行编写代码训练模型,只需要将模型保存到/home/work/PretrainedModel/目录即可。 使用示例代码完成模型训练的步骤如下所示:

          使用示例代码完成模型训练的步骤如下所示:

          1. 在Notebook中执行如下代码查看已导入的数据集。

            ! ls /home/work/data/

          2. 在Notebook中执行train.sh脚本进行模型训练。

            ! bash demo/train.sh /home/work/data/96159/ /home/work/PretrainedModel/

          请根据实际情况修改上述代码中的数据集路径。

          1. 查看生成的模型文件。

            在Notebook左侧导航栏中单击“文件夹”,并打开“Pretrainedmodel”文件夹,如下所示:

            fig003.png

          上一篇
          创建并启动Notebook
          下一篇
          保存Notebook中的模型