提交 0fb10359 编写于 作者: L liweibin

bugfix: naming conflict in graphsage

上级 96cc0ec6
...@@ -195,7 +195,7 @@ def run_epoch(batch_iter, ...@@ -195,7 +195,7 @@ def run_epoch(batch_iter,
if num_trainer > 1: if num_trainer > 1:
num_samples = sum( num_samples = sum(
[len(batch["node_index"]) for batch in batch_feed_dict]) [len(_batch["node_index"]) for _batch in batch_feed_dict])
else: else:
num_samples = len(batch_feed_dict["node_index"]) num_samples = len(batch_feed_dict["node_index"])
total_loss += batch_loss * num_samples total_loss += batch_loss * num_samples
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册