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()
父主题: register接口