fixes

Author: CarloLucibello

docs/Project.toml

@@ -2,6 +2,6 @@
 Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 GraphNeuralNetworks = "cffab07f-9bc2-4db1-8861-388f63bf7694"
-Graphs = "093fc24a-ae57-5d10-9952-331d41423f4d"
+Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"

examples/Project.toml

@@ -4,7 +4,7 @@ DiffEqFlux = "aae7a2af-3d4f-5e19-a356-7da93b79d9d0"
 DifferentialEquations = "0c46a032-eb83-5123-abaf-570d42b7fbaa"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 GraphNeuralNetworks = "cffab07f-9bc2-4db1-8861-388f63bf7694"
-Graphs = "093fc24a-ae57-5d10-9952-331d41423f4d"
+Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 MLDatasets = "eb30cadb-4394-5ae3-aed4-317e484a6458"
 NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 NNlibCUDA = "a00861dc-f156-4864-bf3c-e6376f28a68d"

examples/node_classification_cora.jl

@@ -12,7 +12,7 @@ CUDA.allowscalar(false)
 function eval_loss_accuracy(X, y, ids, model, g)
     ŷ = model(g, X)
     l = logitcrossentropy(ŷ[:,ids], y[:,ids])
-    acc = mean(onecold(ŷ[:,ids] |> cpu) .== onecold(y[:,ids] |> cpu))
+    acc = mean(onecold(ŷ[:,ids]) .== onecold(y[:,ids]))
     return (loss = round(l, digits=4), acc = round(acc*100, digits=2))
 end
 

test/examples/node_classification_cora.jl

@@ -10,7 +10,7 @@ CUDA.allowscalar(false)
 function eval_loss_accuracy(X, y, ids, model, g)
     ŷ = model(g, X)
     l = logitcrossentropy(ŷ[:,ids], y[:,ids])
-    acc = mean(onecold(ŷ[:,ids] |> cpu) .== onecold(y[:,ids] |> cpu))
+    acc = mean(onecold(ŷ[:,ids]) .== onecold(y[:,ids]))
     return (loss = round(l, digits=4), acc = round(acc*100, digits=2))
 end