diff --git a/TensorFlow/built-in/cv/detection/OpenPose_ID0117_for_TensorFlow/Action/training/train_rt.py b/TensorFlow/built-in/cv/detection/OpenPose_ID0117_for_TensorFlow/Action/training/train_rt.py index ecb86722bb2995f281dbf1379c7cfbcee8716002..41b8c071275d0218091a9a7828f3ded762b155c3 100644 --- a/TensorFlow/built-in/cv/detection/OpenPose_ID0117_for_TensorFlow/Action/training/train_rt.py +++ b/TensorFlow/built-in/cv/detection/OpenPose_ID0117_for_TensorFlow/Action/training/train_rt.py @@ -186,6 +186,7 @@ model.add(Dense(units=4, activation='softmax')) # units = nums of classes sess_config = tf.ConfigProto() custom_op = sess_config.graph_options.rewrite_options.custom_optimizers.add() custom_op.name = "NpuOptimizer" +custom_op.parameter_map["jit_compile"].b = True #custom_op.parameter_map["dynamic_input"].b = True #if args.dynamic_input == "lazy_recompile": # custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile") diff --git a/TensorFlow/built-in/cv/image_classification/Face-ResNet50_ID1372_for_TensorFlow/TrainResNet_rt.py b/TensorFlow/built-in/cv/image_classification/Face-ResNet50_ID1372_for_TensorFlow/TrainResNet_rt.py index 976408b01f27d5afffe97fa7fd9efd6e58721398..ecd88cace572382f83957f4b08dffe22e1db6f78 100644 --- a/TensorFlow/built-in/cv/image_classification/Face-ResNet50_ID1372_for_TensorFlow/TrainResNet_rt.py +++ b/TensorFlow/built-in/cv/image_classification/Face-ResNet50_ID1372_for_TensorFlow/TrainResNet_rt.py @@ -144,6 +144,7 @@ def Train(epochs=100): global_config = tf.ConfigProto() custom_op = global_config.graph_options.rewrite_options.custom_optimizers.add() custom_op.name = "NpuOptimizer" + custom_op.parameter_map["jit_compile"].b = True #custom_op.parameter_map["dynamic_input"].b = 1 #custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile") global_config.graph_options.rewrite_options.remapping = RewriterConfig.OFF diff --git a/TensorFlow/built-in/nlp/FastText_ID0135_for_TensorFlow/word_embeddings_rt.py b/TensorFlow/built-in/nlp/FastText_ID0135_for_TensorFlow/word_embeddings_rt.py index 0b87e8a0a488d66f811b780d77c55111b019b05d..da26ed43b42ab3345c1f3c7c8c1c4bab0581dfb1 100644 --- a/TensorFlow/built-in/nlp/FastText_ID0135_for_TensorFlow/word_embeddings_rt.py +++ b/TensorFlow/built-in/nlp/FastText_ID0135_for_TensorFlow/word_embeddings_rt.py @@ -261,6 +261,7 @@ def classification_model(args,embed_dim,X_train_pad,X_test_pad,y_train,y_test,vo sess_config = tf.ConfigProto() custom_op = sess_config.graph_options.rewrite_options.custom_optimizers.add() custom_op.name = "NpuOptimizer" + custom_op.parameter_map["jit_compile"].b = True # custom_op.parameter_map["dynamic_input"].b = True # custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile") custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes(args.precision_mode) diff --git a/TensorFlow/built-in/nlp/Textcnn_ID0123_For_Tensorflow/run_cnn_rt.py b/TensorFlow/built-in/nlp/Textcnn_ID0123_For_Tensorflow/run_cnn_rt.py index f92e36462001d6b7421efc0b59c94c356d9f4b6a..87f19f6d98bf2b5256604fcb51147471c3e5842b 100644 --- a/TensorFlow/built-in/nlp/Textcnn_ID0123_For_Tensorflow/run_cnn_rt.py +++ b/TensorFlow/built-in/nlp/Textcnn_ID0123_For_Tensorflow/run_cnn_rt.py @@ -158,6 +158,7 @@ def train(): custom_op.name = "NpuOptimizer" custom_op.parameter_map["use_off_line"].b = True # 必须显示开启,在昇腾AI处理器执行训练 sess_config.graph_options.rewrite_options.remapping = RewriterConfig.OFF # 必须显示关闭remap + custom_op.parameter_map["jit_compile"].b = True #custom_op.parameter_map["dynamic_input"].b = True custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("allow_mix_precision") #custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile") @@ -297,4 +298,4 @@ if (__name__ == '__main__'): test() else: train() - test() \ No newline at end of file + test() diff --git a/TensorFlow/contrib/nlp/texting/texting_ID0193_for_TensorFlow/train_rt.py b/TensorFlow/contrib/nlp/texting/texting_ID0193_for_TensorFlow/train_rt.py index 837d0442dfe23899d84466d80254334d89383038..d974e1216a9c27e40ce8813b7bcc91aabd03fb1d 100644 --- a/TensorFlow/contrib/nlp/texting/texting_ID0193_for_TensorFlow/train_rt.py +++ b/TensorFlow/contrib/nlp/texting/texting_ID0193_for_TensorFlow/train_rt.py @@ -44,6 +44,7 @@ config = tf.ConfigProto() 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["jit_compile"].b =True #custom_op.parameter_map["dynamic_input"].b =True #custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile") config.graph_options.rewrite_options.remapping = RewriterConfig.OFF