From 3d23852c2fd753364b19c316ded7c4b41805d47a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=AE=9D=E5=BC=9F?= Date: Wed, 13 Apr 2022 01:08:50 +0000 Subject: [PATCH] update --- .../src/tools_pix2pose/3_train_pix2pose.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 06a24c39f..558791c47 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 -- Gitee