GraphModifierHook

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

参数名	参数类型	参数说明	必选/可选
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对象

父主题： 类参考