From da69d6ded40821e09b3ed5c8a27f16f3dd14646e Mon Sep 17 00:00:00 2001 From: huangju1993 Date: Tue, 29 Mar 2022 03:07:13 +0000 Subject: [PATCH] update TensorFlow/contrib/cv/VDSR_ID2114_for_TensorFlow/VDSR.py. --- TensorFlow/contrib/cv/VDSR_ID2114_for_TensorFlow/VDSR.py | 1 + 1 file changed, 1 insertion(+) diff --git a/TensorFlow/contrib/cv/VDSR_ID2114_for_TensorFlow/VDSR.py b/TensorFlow/contrib/cv/VDSR_ID2114_for_TensorFlow/VDSR.py index 5c44dd5d0..0e32f291c 100644 --- a/TensorFlow/contrib/cv/VDSR_ID2114_for_TensorFlow/VDSR.py +++ b/TensorFlow/contrib/cv/VDSR_ID2114_for_TensorFlow/VDSR.py @@ -185,6 +185,7 @@ if __name__ == '__main__': custom_op = config.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") + custom_op.parameter_map["auto_tune_mode"].s = tf.compat.as_bytes("RL,GA") config.graph_options.rewrite_options.remapping = RewriterConfig.OFF # 必须显式关闭remap # custom_op.parameter_map["use_off_line"].b = True # config.graph_options.rewrite_options.memory_optimization = RewriterConfig.OFF -- Gitee