fix epoch loss (#6319)

3ded3587 · SuperYY · GitHub · 13e7c2fa · 3ded3587
Unverified Commit 3ded3587 authored Feb 16, 2024 by SuperYY Committed by GitHub Feb 16, 2024
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

examples/pytorch/mvgrl/graph/main.py examples/pytorch/mvgrl/graph/main.py +2 -2

No files found.
--- a/examples/pytorch/mvgrl/graph/main.py
+++ b/examples/pytorch/mvgrl/graph/main.py
@@ -130,8 +130,8 @@ if __name__ == "__main__":

        print("Epoch {}, Loss {:.4f}".format(epoch, loss_all))

-        if loss < best:
-            best = loss
+        if loss_all < best:
+            best = loss_all
            best_t = epoch
            cnt_wait = 0
            th.save(model.state_dict(), f"{args.dataname}.pkl")