CarloLucibello / GraphNeuralNetworks.jl

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

Add `GConvGRU` temporal layer #438

Closed aurorarossi closed 1 month ago

aurorarossi commented 1 month ago

This PR adds the GConvGRU temporal convolutional layer of the following paper https://arxiv.org/pdf/1612.07659 I tested the training also on GPU.