MindStudio IDE执行模型训练的主要步骤包括:
当前支持以MindSpore、TensorFlow和PyTorch训练框架为模板创建训练工程,其中MindSpore训练框架可以使用MindSpore Insight实现训练过程可视化,MindSpore Insight详细使用方法请参见MindSpore Insight官方文档。
参数 |
说明 |
---|---|
Project name |
工程名称,自行配置。
|
Project location |
工程默认保存路径,用户可自定义。(对于首次使用MindStudio IDE的用户,该项默认为“$HOME/MindstudioProjects”。) |
More settings |
“Module name”:模块名,默认与“Project name”一致。 |
“Content root”:根目录下路径。 |
|
“Module file location”:模块文件路径。 |
|
单击“Project format”右侧选框,出现下拉菜单。
|
├── .idea ├── data //数据集目录,需用自行创建 ├── .project //工程信息文件,包含工程类型、工程描述、运行目标设备类型以及CANN版本 ├── train.py //训练脚本文件,为空文件,用户在这里编写训练脚本 ├── MyTraining.iml
如果已有训练工程,则无需新建训练工程,可直接通过MindStudio IDE导入,操作如下。
运行配置支持Ascend Training和Python两种配置方式(推荐使用Python配置方式):
参数 |
说明 |
---|---|
Name |
用户自行定义。 |
Script path |
选择需要执行调试的Python源文件具体路径。 |
Parameters |
运行参数,请根据实际情况自行配置。 |
Python interpreter |
“Use SDK of module”:使用模块级的Python SDK进行解析。 具体配置功能请参见设置模块级Python SDK(昇腾工程)或设置模块级Python SDK(非昇腾工程)。 |
“Use specified interpreter”:使用已配置的特定解析器。 具体配置功能请参见Python SDK设置。 |
|
Working directory |
工作目录(默认为需执行调试的Python源文件所在的目录)。 |