Dear developers and experts,
I am trying to program with Flux.jl
for a GNN project. There is a simple question: can I upload a model, form example, GraphConv
in GeometricFlux.jl
, to GPU via |> gpu ? I tried this but did not observed the expected behaviour. Can anyone help ?