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 a4a94ca5c3e2a17cdb907aeb4eb6d22a48e193e7..862f8a95766c33d41a709069255d4e55e401f06f 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
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 1af7f77f65cba4998449eb120231a2ac707b2ed1..7f3a0cfed00950e13a30538af60d270d313ca30a 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,