diff --git a/TensorFlow/contrib/cv/VDSR_ID2114_for_TensorFlow/VDSR.py b/TensorFlow/contrib/cv/VDSR_ID2114_for_TensorFlow/VDSR.py index 5c44dd5d05892122d2531e634f5d18ae384a7cd0..0e32f291cdc7f2a1ac49a1b0b6678e4a435a3054 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