From a8d3100011a512b3ae7eded8be60d268f73f412e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E6=B4=8B=E6=B4=8B?= <10527367+yang-yang-zhang123456@user.noreply.gitee.com> Date: Thu, 7 Apr 2022 07:49:40 +0000 Subject: [PATCH] update TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py. --- TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py b/TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py index a02c70345..b8893c80b 100644 --- a/TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py +++ b/TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py @@ -233,13 +233,13 @@ def main(argv=None): iterator = dataset.make_one_shot_iterator() images0, agelabels0, genderlabels0 = iterator.get_next() for step in range(num_steps): - start_time = time.time() + #start_time = time.time() images, agelabels_1, genderlabels_1 = sess.run([images0, agelabels0, genderlabels0]) # images1 = (np.reshape(images,(FLAGS.batch_size,227,227,3))).astype(np.float32) agelabels_1 = (np.reshape(agelabels_1, (FLAGS.batch_size))).astype(np.int32) genderlabels_1 = (np.reshape(genderlabels_1, (FLAGS.batch_size))).astype(np.int32) - + start_time = time.time() _,totallossvalue, agelossvalue, genderlossvalue = sess.run([ agegendertrain_op, totallosses, agelosses, genderlosses], feed_dict = {images_holder:images,agelabels_holder: agelabels_1, genderlabels_holder:genderlabels_1}) duration = time.time() - start_time -- Gitee