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 02f4dced17a64c4e77bfed9683f221611662ee6d..70c0a4de7fcc71d18928166e4106e2cb136a4734 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)