From 2778738f8ba72b0b79c1400cb04a8dff5ab0d237 Mon Sep 17 00:00:00 2001 From: zhengying Date: Thu, 29 Dec 2022 11:13:51 +0800 Subject: [PATCH] =?UTF-8?q?MobileNetV1=5FID0093=5Ffor=5FTensorFlow?= =?UTF-8?q?=E7=BD=91=E7=BB=9C=E7=9A=84=E5=9B=BE=E4=B8=AD=E7=AE=97=E5=AD=90?= =?UTF-8?q?=E6=89=A7=E8=A1=8C=E6=94=B9=E4=B8=BABFS=E6=96=B9=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../MobileNetV1_ID0093_for_TensorFlow/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/TensorFlow/built-in/cv/image_classification/MobileNetV1_ID0093_for_TensorFlow/train.py b/TensorFlow/built-in/cv/image_classification/MobileNetV1_ID0093_for_TensorFlow/train.py index 4c4945735..9a057c1e7 100644 --- a/TensorFlow/built-in/cv/image_classification/MobileNetV1_ID0093_for_TensorFlow/train.py +++ b/TensorFlow/built-in/cv/image_classification/MobileNetV1_ID0093_for_TensorFlow/train.py @@ -126,6 +126,7 @@ def main(args): custom_op.name = "NpuOptimizer" custom_op.parameter_map["use_off_line"].b = True custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes(args.precision_mode) + custom_op.parameter_map["graph_run_mode"].i = 0 config_proto.graph_options.rewrite_options.remapping = RewriterConfig.OFF sess = tf.Session(config=npu_config_proto(config_proto=config_proto)) sess.run(tf.global_variables_initializer()) -- Gitee