更新时间:2024-10-30 GMT+08:00
模型文件基本要求
模型文件通常包括模型图和模型权重文件,具体内容视用户算法决定,无其他要求。如果需要使用内置指标计算,则需满足以下条件:
自定义模型包中必须包含推理启动文件以及自定义模型文件,其中推理启动文件需要根据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 #自定义模型文件(必选,名称可自定义) └─ ... #其他(可选)
父主题: 模型文件说明(训练)