From 326710993fb2f7560c4091a5883e21e1083d249e Mon Sep 17 00:00:00 2001 From: William L Hamilton Date: Fri, 3 Nov 2017 12:19:52 -0700 Subject: [PATCH] Removed extra scaling by batch size. --- graphsage/models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/graphsage/models.py b/graphsage/models.py index e9fe791..e3a904f 100644 --- a/graphsage/models.py +++ b/graphsage/models.py @@ -388,7 +388,6 @@ class SampleAndAggregate(GeneralizedModel): self.loss += FLAGS.weight_decay * tf.nn.l2_loss(var) self.loss = self.link_pred_layer.loss(self.outputs1, self.outputs2, self.neg_outputs) - self.loss = self.loss / tf.cast(self.batch_size, tf.float32) tf.summary.scalar('loss', self.loss) def _accuracy(self):