From 40cb2509ea39ade24a4f2febd024dac00bf041f1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=99=86=E6=96=87=E9=9C=9E?= <10511946+lu-wenxia@user.noreply.gitee.com> Date: Fri, 15 Apr 2022 02:03:36 +0000 Subject: [PATCH 1/2] update --- .../cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/main.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/TensorFlow/contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/main.py b/TensorFlow/contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/main.py index 1af7f77f6..7f3a0cfed 100644 --- a/TensorFlow/contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/main.py +++ b/TensorFlow/contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/main.py @@ -75,6 +75,12 @@ def main(_): os.makedirs(FLAGS.checkpoint_dir) if not os.path.exists(FLAGS.sample_dir): os.makedirs(FLAGS.sample_dir) + + config_proto = tf.ConfigProto() + custom_op = config_proto.graph_options.rewrite_options.custom_optimizers.add() + custom_op.name = 'NpuOptimizer' + custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("allow_mix_precision") + config = npu_config_proto(config_proto=config_proto) with tf.Session(config=config) as sess: srcnn = CGAN(sess, -- Gitee From a856b77359ec9e92f6619d1cf5ea653cebb8d5a2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=99=86=E6=96=87=E9=9C=9E?= <10511946+lu-wenxia@user.noreply.gitee.com> Date: Fri, 15 Apr 2022 02:40:23 +0000 Subject: [PATCH 2/2] update --- .../contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/cfg.py | 1 + 1 file changed, 1 insertion(+) diff --git a/TensorFlow/contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/cfg.py b/TensorFlow/contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/cfg.py index a4a94ca5c..862f8a957 100644 --- a/TensorFlow/contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/cfg.py +++ b/TensorFlow/contrib/cv/fusiongan/FusionGAN_ID2124_for_TensorFlow/cfg.py @@ -39,6 +39,7 @@ def make_config(FLAGS): custom_op = config.graph_options.rewrite_options.custom_optimizers.add() custom_op.name = "NpuOptimizer" custom_op.parameter_map["use_off_line"].b = True + custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("allow_mix_precision") config.graph_options.rewrite_options.remapping = RewriterConfig.OFF ## Auto Tune -- Gitee