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对象
父主题： 类参考