更新时间:2024-03-06 GMT+08:00

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”目录下。