更新时间:2024-03-06 GMT+08:00
PyTorch-py27通用模板
简介
搭载PyTorch1.0AI引擎,运行环境为“python2.7”,内置输入输出模式为未定义模式,请根据模型功能或业务场景重新选择合适的输入输出模式。使用该模板导入模型时请选择到包含模型文件的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”目录下。
父主题: 模板说明