From 5815dc04d20e957a1e6aeaf9393f5bd2a28d6eda Mon Sep 17 00:00:00 2001 From: huangju1993 Date: Fri, 17 Nov 2023 01:15:39 +0000 Subject: [PATCH 1/2] update TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/main.py. Signed-off-by: huangju1993 --- TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/main.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/main.py b/TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/main.py index 5f7c2f953..c80a2ad73 100644 --- a/TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/main.py +++ b/TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/main.py @@ -40,6 +40,8 @@ from npu_bridge.estimator import npu_ops from tensorflow.core.protobuf.rewriter_config_pb2 import RewriterConfig from npu_bridge.npu_init import * +from npu_bridge.estimator.npu import npu_plugin +npu_plugin.set_device_sat_mode(1) flags = tf.app.flags flags.DEFINE_integer("epoch", 25, "Epoch to train [25]") -- Gitee From e3f4eac1434e159038e66b0aebc63969d700e51b Mon Sep 17 00:00:00 2001 From: huangju1993 Date: Fri, 17 Nov 2023 01:18:40 +0000 Subject: [PATCH 2/2] update TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/model.py. Signed-off-by: huangju1993 --- .../DCGAN_ID0686_for_TensorFlow/model.py | 32 +++++++++---------- 1 file changed, 16 insertions(+), 16 deletions(-) diff --git a/TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/model.py b/TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/model.py index 7e3575453..e22a821a1 100644 --- a/TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/model.py +++ b/TensorFlow/built-in/graph/DCGAN_ID0686_for_TensorFlow/model.py @@ -318,22 +318,22 @@ class DCGAN(object): # modify for NPU start # 增加loss scale打印 lossScale = tf.get_default_graph().get_tensor_by_name("loss_scale:0") - overflow_status_reduce_all = tf.get_default_graph().get_tensor_by_name("overflow_status_reduce_all:0") - - # _, summary_str = self.sess.run([d_optim, self.d_sum], - # feed_dict={ - # self.inputs: batch_images, - # self.z: batch_z, - # self.y:batch_labels, - # }) - l_s, overflow_status_reduce_all, _, summary_str = self.sess.run( - [lossScale, overflow_status_reduce_all, d_optim, self.d_sum], feed_dict={ - self.inputs: batch_images, - self.z: batch_z, - self.y:batch_labels, - }) - print("lossScale==========================:", l_s) - print("overflow_status_reduce_all is======:", overflow_status_reduce_all) + #overflow_status_reduce_all = tf.get_default_graph().get_tensor_by_name("overflow_status_reduce_all:0") + + _, summary_str = self.sess.run([d_optim, self.d_sum], + feed_dict={ + self.inputs: batch_images, + self.z: batch_z, + self.y:batch_labels, + }) + #l_s, overflow_status_reduce_all, _, summary_str = self.sess.run( + # [lossScale, overflow_status_reduce_all, d_optim, self.d_sum], feed_dict={ + # self.inputs: batch_images, + # self.z: batch_z, + # self.y:batch_labels, + # }) + #print("lossScale==========================:", l_s) + #print("overflow_status_reduce_all is======:", overflow_status_reduce_all) # 注释add summary提升性能 # self.writer.add_summary(summary_str, counter) # modify for NPU end -- Gitee