diff --git a/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/src/tools_pix2pose/3_train_pix2pose.py b/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/src/tools_pix2pose/3_train_pix2pose.py index 06a24c39ffef5cc1ca2a498935e62cd49f3d9c95..558791c47821375705468a0cf398862049ef9316 100644 --- a/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/src/tools_pix2pose/3_train_pix2pose.py +++ b/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/src/tools_pix2pose/3_train_pix2pose.py @@ -92,7 +92,8 @@ from tools_pix2pose import bop_io sess_config = tf.ConfigProto() custom_op = sess_config.graph_options.rewrite_options.custom_optimizers.add() -custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("force_fp32") +# custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("force_fp32") +custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("allow_mix_precision") custom_op.name = "NpuOptimizer" sess_config.graph_options.rewrite_options.remapping = RewriterConfig.OFF sess_config.graph_options.rewrite_options.memory_optimization = RewriterConfig.OFF