Skip to content

GitLab

Unverified Commit 3ded3587 authored by SuperYY's avatar SuperYY Committed by GitHub
Browse files

fix epoch loss (#6319)

parent 13e7c2fa
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/pytorch/mvgrl/graph/main.py
View file @ 3ded3587
...@@ -130,8 +130,8 @@ if __name__ == "__main__": ...@@ -130,8 +130,8 @@ if __name__ == "__main__":
print("Epoch {}, Loss {:.4f}".format(epoch, loss_all)) print("Epoch {}, Loss {:.4f}".format(epoch, loss_all))
if loss < best: if loss_all < best:
best = loss best = loss_all
best_t = epoch best_t = epoch
cnt_wait = 0 cnt_wait = 0
th.save(model.state_dict(), f"{args.dataname}.pkl") th.save(model.state_dict(), f"{args.dataname}.pkl")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment