GraphModifierHook
自动改图Hook类，仅在使用NPUEstimator训练模式下使用，添加后即可开启自动改图功能。
参数名 |
参数类型 |
参数说明 |
|---|---|---|
dump_graph |
bool |
是否保存TensorFlow当前计算图，默认为“True”。 |
modify_graph |
bool |
是否开启自动改图功能，默认为“True”。 |
使用示例
from mx_rec.graph.modifier import GraphModifierHook est.train(input_fn=lambda: input_fn(), hooks=[GraphModifierHook()])
父主题： 类参考
