From 425bb528794dcdc9652e6be4ab7a38115f7685c1 Mon Sep 17 00:00:00 2001 From: Namshik Kim Date: Sat, 11 Nov 2017 02:05:11 -0800 Subject: [PATCH] Update train.py change from version up --- cs20si/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cs20si/train.py b/cs20si/train.py index f8f8f59..99c48c4 100644 --- a/cs20si/train.py +++ b/cs20si/train.py @@ -11,7 +11,7 @@ num_epochs = 50 def calculate_loss(original, reconstructed): - return tf.div(tf.reduce_sum(tf.square(tf.sub(reconstructed, + return tf.div(tf.reduce_sum(tf.square(tf.subtract(reconstructed, original))), tf.constant(float(batch_size)))