昇腾社区首页
中文
注册

def compile_pipe

函数功能

Pipeline优化模块,内置Pipeline调优能力,目前仅支持对MindIE SD中内置Pipeline优化,该函数在MindIE SD仓的路径为:mindiesd/pipeline/compile_pipe.py。

函数原型

def compile_pipe(pipe, cfg=None):

参数说明

参数名

输入/输出

类型

说明

pipe

输入

module

MindIE SD中提供的pipeline类。

cfg

输入

-

pipeline优化配置。

返回值说明

返回优化后的pipeline对象。