diff --git a/TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py b/TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py index 19fc2bd44e115dcf32ada1363ced05f313587856..b8893c80bd6a042120f0d1828905d7cf779e7072 100644 --- a/TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py +++ b/TensorFlow/contrib/cv/LMTCNN_ID1278_for_TensorFlow/train.py @@ -239,7 +239,7 @@ def main(argv=None): # 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() + 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