search

CarloLucibello / GraphNeuralNetworks.jl

Graph Neural Networks in Julia
https://carlolucibello.github.io/GraphNeuralNetworks.jl/dev/
MIT License
220 stars 46 forks source link

Add `Flux.gpu` function for `TemporalSnapshotsGNNGraph` type #362

Closed aurorarossi closed 8 months ago

CarloLucibello commented 8 months ago

Did you run the gpu test locally to see if it passes?

aurorarossi commented 8 months ago

Yes!