GraphModifierHook

自动改图Hook类,仅在使用Estimator训练模式下使用,添加后即可开启自动改图功能。

参数名

类型

必选/可选

说明

dump_graph

bool

可选

是否保存TensorFlow当前计算图,默认为“False”

modify_graph

bool

可选

是否开启自动改图功能,默认为“True”

使用示例

from mx_rec.graph.modifier import GraphModifierHook

#定义数据处理函数
def input_fn():
     pass

est.train(input_fn=lambda: input_fn(), hooks=[GraphModifierHook()])   #est为创建的NPUEstimator对象