diff --git a/TensorFlow/contrib/cv/Milking_cowmask_ID0712_for_TensorFlow/architectures/model.py b/TensorFlow/contrib/cv/Milking_cowmask_ID0712_for_TensorFlow/architectures/model.py
index 8d1174300ffdf7dc0d2221ec09d93c309ad0514e..967ddc654bc1f0a4011dd926db2127f9d1d55804 100644
--- a/TensorFlow/contrib/cv/Milking_cowmask_ID0712_for_TensorFlow/architectures/model.py
+++ b/TensorFlow/contrib/cv/Milking_cowmask_ID0712_for_TensorFlow/architectures/model.py
@@ -147,7 +147,13 @@ class Model():
         self.seed = 123
         self.create_net()
         self.saver = tf.train.Saver()
-        self.sess = tf.Session(config=npu_config_proto())
+        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)
+        self.sess = tf.Session(config=config)
+        # self.sess = tf.Session(config=npu_config_proto())
         self.metrics['init'] = tf.global_variables_initializer()
         if ckpt:
             model_file = tf.train.latest_checkpoint(self.metrics['model_path'])