AI开发平台MODELARTS-创建算法:设置算法启动方式(预置框架)

时间:2024-10-15 14:28:15

设置算法启动方式(预置框架)

选择“预置框架”创建算法。

图1 使用预置框架创建算法
用户需根据实际算法代码情况设置“镜像”“代码目录”“启动文件”。选择的AI镜像和编写算法代码时选择的框架必须一致。例如编写算法代码使用的是TensorFlow,则在创建算法时也要选择TensorFlow镜像。
表1 启动方式参数说明

参数

说明

“启动方式”

选择“预置框架”

选择算法使用的预置框架引擎和引擎版本。

“代码目录”

算法代码存储的OBS路径。训练代码、依赖安装包或者预生成模型等训练所需文件上传至该代码目录下。

请注意不要将训练数据放在代码目录路径下。训练数据比较大,训练代码目录在训练作业启动后会下载至后台,可能会有下载失败的风险。

训练作业创建完成后,ModelArts会将代码目录及其子目录下载至训练后台容器中。

例如:OBS路径“obs://obs-bucket/training-test/demo-code”作为代码目录,OBS路径下的内容会被自动下载至训练容器的“${MA_JOB_DIR}/demo-code”目录中,demo-code为OBS存放代码路径的最后一级目录,用户可以根据实际修改。

说明:
  • 编程语言不限。
  • 文件数(含文件、文件夹数量)小于或等于1000个。
  • 文件总大小小于或等于5GB。

“启动文件”

必须为“代码目录”下的文件,且以“.py”结尾,即ModelArts目前只支持使用Python语言编写的启动文件。

代码目录路径中的启动文件为训练启动的入口。

support.huaweicloud.com/usermanual-standard-modelarts/develop-modelarts-0003.html