From 0fb1035908447be2d12291239aef993030db8e10 Mon Sep 17 00:00:00 2001 From: liweibin Date: Mon, 20 Jan 2020 17:00:17 +0800 Subject: [PATCH] bugfix: naming conflict in graphsage --- examples/graphsage/train_multi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/graphsage/train_multi.py b/examples/graphsage/train_multi.py index 8a1e799..6655db9 100644 --- a/examples/graphsage/train_multi.py +++ b/examples/graphsage/train_multi.py @@ -195,7 +195,7 @@ def run_epoch(batch_iter, if num_trainer > 1: num_samples = sum( - [len(batch["node_index"]) for batch in batch_feed_dict]) + [len(_batch["node_index"]) for _batch in batch_feed_dict]) else: num_samples = len(batch_feed_dict["node_index"]) total_loss += batch_loss * num_samples -- GitLab