华为云AI开发平台ModelArtsPyTorch-py36通用模板_云淘科技

简介

搭载PyTorch1.0AI引擎,运行环境为“python3.6”,内置输入输出模式为未定义模式,请根据模型功能或业务场景重新选择合适的输入输出模式。使用该模板导入模型时请选择到包含模型文件的model目录。

模板输入

存储在OBS上的PyTorch模型包,确保您使用的OBS目录与ModelArts在同一区域。模型包的要求请参见模型包示例。

对应的输入输出模式

未定义模式,可覆盖,即创建模型时支持选择其他输入输出模式。

模型包规范

模型包必须存储在OBS中,且必须以“model”命名。“model”文件夹下面放置模型文件、模型推理代码。
模型推理代码文件必选,其文件名必须为“customize_service.py”,“model”文件夹下有且只能有1个推理代码文件,模型推理代码编写请参见模型推理代码编写说明。

使用模板导入的模型包结构如下所示:

model/
│
├── 模型文件                 //必选,不同的框架,其模型文件格式不同,详细可参考模型包示例。
├── 自定义Python包           //可选,用户自有的Python包,在模型推理代码中可以直接引用。
├── customize_service.py  //必选,模型推理代码,文件名称必须为“customize_service.py”,否则不视为推理代码。

模型包示例

PyTorch模型包结构

发布该模型时只需要指定到“model”目录。

OBS桶/目录名
|── model    必选,文件夹名称必须为“model”,用于放置模型相关文件。
   ├── <>     可选,用户自有的Python包,在模型推理代码中可以直接引用。
   ├── resnet50.pth           必选,pytorch模型保存文件,存有权重变量等信息。
   ├──customize_service.py    必选,模型推理代码,文件名称必须为“customize_service.py”, 有且只有1个推理代码文件。“customize_service.py”依赖的“py”文件可以直接放“model”目录下。

父主题: 模板说明

同意关联代理商云淘科技,购买华为云产品更优惠(QQ 78315851)

内容没看懂? 不太想学习?想快速解决? 有偿解决: 联系专家