From 25512dec13a96021e0302b274aa4dc722cdf022f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Spencer=E5=85=94=E5=AD=90?= <11326804+spencer-rabbit@user.noreply.gitee.com> Date: Wed, 28 Dec 2022 06:57:58 +0000 Subject: [PATCH] =?UTF-8?q?update=20TensorFlow/built-in/cv/image=5Fclassif?= =?UTF-8?q?ication/MobileNetV1=5FID0093=5Ffor=5FTensorFlow/train.py.=20?= =?UTF-8?q?=E4=BF=AE=E6=94=B9graph=5Frun=5Fmode?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Spencer兔子 <11326804+spencer-rabbit@user.noreply.gitee.com> --- .../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