update output display

Lolash · Mar 23, 2019 · 88aa788 · 88aa788
1 parent e9bf240
commit 88aa788
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/src/__pycache__/utils.cpython-35.pyc b/src/__pycache__/utils.cpython-35.pyc
diff --git a/src/utils.py b/src/utils.py
@@ -79,7 +79,7 @@ def apply_model(dataCenter, ds, graphSage, classification, unsupervised_loss, b_
 		logists = classification(embs_batch)
 		loss = -torch.sum(logists[range(logists.size(0)), labels_batch], 0)
 		loss /= len(nodes_batch)
-		print('Step {}, Loss: {:.4f}, Dealed Nodes [{}/{}] '.format(index, loss.item(), len(visited_nodes), len(train_nodes)))
+		print('Step [{}/{}], Loss: {:.4f}, Dealed Nodes [{}/{}] '.format(index, batches, loss.item(), len(visited_nodes), len(train_nodes)))
 		loss.backward()
 		for model in models:
 			nn.utils.clip_grad_norm_(model.parameters(), 5)