fix learn it

TuGraph-family · Dec 26, 2023 · 99dd1b9 · 99dd1b9
1 parent 9a9cc55
commit 99dd1b9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/learn/examples/train_full_mag.py b/learn/examples/train_full_mag.py
@@ -403,7 +403,8 @@ def train(
 
         pbar.close()
         loss = total_loss / num_train
-
+        if epoch == 4 and loss < 5:
+            print("The loss value is less than 5")
         result = test(g, model, node_embed, labels, device, split_idx)
         logger.add_result(run, result)
         train_acc, valid_acc, test_acc = result