自动驾驶云服务 Octopus
自动驾驶云服务 Octopus
- 功能总览
- 产品介绍
- 快速入门
- 用户指南
- API参考
- SDK参考
- 常见问题
- 文档下载
- 通用参考
本文导读
展开导读
链接复制成功！
模型文件说明（训练）
Octopus模型管理模块，支持用户上传模型，并将其用于模型评测、模型编译任务。如果需要将模型用于内置评测模板评测，除模型文件外，需另外包含推理启动文件：
- customer_inference.py
仅当需要使用内置评测指标计算时需要添加推理启动文件，文件名称可自定义，将该文件置于模型目录下。
为了显示以上文件的使用流程，可参考如下图示：
图1 使用流程
模型文件基本要求
模型文件通常包括模型图和模型权重文件，具体内容视用户算法决定，无其他要求。如果需要使用内置指标计算，则需满足以下条件：
自定义模型包中必须包含推理启动文件以及自定义模型文件，其中推理启动文件需要根据Octopus数据格式对推理结果存储为json进行适配，除此之外还可以包含一些其他必要的自定义脚本、自定义库、自定义算子库等。模型包文件目录结构可参考如下：
.
├─ model #模型包根目录，上传模型时进入该目录下（名称可自定义）
├─ customer_inference.py #“模型推理”启动文件（如该模型用于内置评测模板评测，需要提供该文件，名称可自定义）
├─ customer_package #自定义库（可选，名称可自定义）
├─ __init__.py #自定义库 - > python库
├─ tensor_define.py #自定义库 - > python库文件
└─ ...
├─ customer_utils.py #自定义脚本1（可选，名称可自定义）
├─ customer_script2.sh #自定义脚本2（可选，名称可自定义）
├─ sub_directory #子文件夹
├─ customer_operation.so #自定义算子库（可选，名称可自定义）
└─ coustomer_model.pb #自定义模型文件（必选，名称可自定义）
└─ ... #其他（可选）
父主题： 模型管理
