昇腾社区首页
中文
注册

BaseRelease类

release_model接口功能

需要用户实现的模型发布接口。

接口引用路径

ockics.modules.estimator.BaseRelease

release_model接口格式

函数:release_model(self, src_train_model_url, src_infer_model_url)

release_model接口输入参数

参数

类型

是否必选

说明

取值要求

src_train_model

str

必选

待发布的训练模型本地路径。

-

src_infer_model_url

str

必选

待发布的推理模型本地路径。

-

release_model接口返回值

release_model接口使用样例

from ockics.common import ClassFactory, ClassType
from ockics.common import LOGGER
from ockics.entry import main_releasemodel
from ockics.modules.estimator import BaseRelease

@ClassFactory.register(ClassType.MODEL_RELEASE)
class MyReleaser(BaseRelease):
    def release_model(self, src_train_model, src_infer_model):
        LOGGER.info("my releaser: %s", src_train_model)
        LOGGER.info("my releaser: %s", src_infer_model)
if __name__ == "__main__":
    main_releasemodel()