Skip to content

GitLab

Unverified Commit 4bf70f09 authored by Harsh Sinha's avatar Harsh Sinha Committed by GitHub
Browse files

[Bugfix] issue #3527 (#3528) 

* Fix issue 3527

* Changed default device

* Added g to device
parent 8dbf3a47
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/pytorch/model_zoo/geometric/mnist.py
View file @ 4bf70f09
...@@ -177,6 +177,7 @@ for epoch in range(10): ...@@ -177,6 +177,7 @@ for epoch in range(10):
for g, x, y in test_loader: for g, x, y in test_loader:
x = x.to(device) x = x.to(device)
y = y.to(device) y = y.to(device)
g = [g_i.to(device) for g_i in g]
out = model(g, x) out = model(g, x)
hit += (out.max(-1)[1] == y).sum().item() hit += (out.max(-1)[1] == y).sum().item()
tot += len(y) tot += len(y)
......
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