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