[Triton课程笔记] 1.2.4 编程实战-配置ensemble model
一、例子1
![](https://img2024.cnblogs.com/blog/741682/202406/741682-20240626141212949-1408631656.jpg)
图片-预处理模型-分别进入不同的模型-输出不同的结果。
![](https://img2024.cnblogs.com/blog/741682/202406/741682-20240626141231722-154030087.jpg)
定义模型的输入输出,然后在ensemble_scheduling中定义不同的步骤,其中step中的key是本身的input/output tensor的名字;value是ensemble model中的Tensor名字。
配置写完后,在ensemble_model的目录只能够新建一个版本目录,里面为空,然后放config文件。
注意事项:
![](https://img2024.cnblogs.com/blog/741682/202406/741682-20240626141256105-1489397105.jpg)
- 如果组合里有一个是stateful 模型,那么整个pipeline都成为stateful 模型,推理请求需要符合stateful model的规则。
- 每个子模块有各自的调度器。
- 如果每个子模块都是framework backend,则传输使用GPU进行,否则可能通过cpu 内存。
二、例子2
![](https://img2024.cnblogs.com/blog/741682/202406/741682-20240626141316569-1930701614.jpg)
语音识别流程。
三、例子3
![](https://img2024.cnblogs.com/blog/741682/202406/741682-20240626141339092-602533804.jpg)
预处理-分支1分割模型-分支2fcn分割-合并拼接
本文版权归作者(https://www.cnblogs.com/harrymore/)和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文链接,如有问题, 可邮件(harrymore@126.com)咨询.