From 9af9456aa9e8eef738554853f81b1fb7951bf466 Mon Sep 17 00:00:00 2001 From: huangju1993 Date: Tue, 20 Sep 2022 09:11:51 +0000 Subject: [PATCH] update TensorFlow/built-in/cv/image_classification/HRNet_ID0139_for_TensorFlow/top/train.py. Signed-off-by: huangju1993 --- .../HRNet_ID0139_for_TensorFlow/top/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/TensorFlow/built-in/cv/image_classification/HRNet_ID0139_for_TensorFlow/top/train.py b/TensorFlow/built-in/cv/image_classification/HRNet_ID0139_for_TensorFlow/top/train.py index 02f4dced1..70c0a4de7 100644 --- a/TensorFlow/built-in/cv/image_classification/HRNet_ID0139_for_TensorFlow/top/train.py +++ b/TensorFlow/built-in/cv/image_classification/HRNet_ID0139_for_TensorFlow/top/train.py @@ -70,6 +70,7 @@ def main(unused_argv): custom_op.name = "NpuOptimizer" custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes(FLAGS.precision_mode) custom_op.parameter_map["use_off_line"].b = True + custom_op.parameter_map["op_debug_config"].s = tf.compat.as_bytes("/npu/hwx1148384/support_check_oom") config.graph_options.rewrite_options.remapping = RewriterConfig.OFF npu_init = npu_ops.initialize_system() sess = tf.Session(config=config) -- Gitee