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接口