gpu working

Author: CarloLucibello

examples/Project.toml

@@ -6,6 +6,5 @@ Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 GraphNeuralNetworks = "cffab07f-9bc2-4db1-8861-388f63bf7694"
 LightGraphs = "093fc24a-ae57-5d10-9952-331d41423f4d"
 MLDatasets = "eb30cadb-4394-5ae3-aed4-317e484a6458"
-ModelingToolkit = "961ee093-0014-501f-94e3-6117800e7a78"
 NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 NNlibCUDA = "a00861dc-f156-4864-bf3c-e6376f28a68d"

examples/neural_ode.jl

@@ -36,10 +36,10 @@ node = NeuralODE(
 )
 
 model = GNNChain(GCNConv(nin => nhidden, relu),
-                Dropout(0.5),
-                node,
-                diffeqarray_to_array,
-                GCNConv(nhidden => nout))
+                 Dropout(0.5),
+                 node,
+                 diffeqarray_to_array,
+                 GCNConv(nhidden => nout))
 
 # Loss
 loss(x, y) = logitcrossentropy(model(g, x), y)
@@ -55,6 +55,6 @@ opt = ADAM(0.01)
 ## Training Loop
 for epoch in 1:epochs
     gs = gradient(() -> loss(X, y), ps)
-    Flux.Optimisers.update!(opt, ps, gs)
+    Flux.Optimise.update!(opt, ps, gs)
     @show(accuracy(X, y))
 end