decompose_graph
功能说明
分解训练代码中的graph用于对分解后的模型进行finetune(微调)。
约束说明
- 用户已经使用auto_decomposition接口成功分解模型。
- 需基于训练代码使用该接口,且必须保证auto_decomposition接口分解的模型和权值是基于该训练代码训练得到的。
函数原型
add_ops = decompose_graph(save_path, graph=None)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
---|---|---|---|
save_path |
输入 |
auto_decomposition接口中的保存路径。 |
数据类型:string |
graph |
输入 |
待分解的graph,可选。如果不输入,则会自动调用当前模型graph。 默认为None。 |
数据类型:tf graph |
add_ops |
返回值 |
模型分解后新添加卷积层的名字的列表。 |
数据类型:list |
返回值说明
模型分解后新添加卷积层的名字的列表。
函数输出
无。
调用示例
from amct_tensorflow.tensor_decompose import decompose_graph decompose_graph(save_path)