search

CarloLucibello / GraphNeuralNetworks.jl

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

Creating dropout functionality in the GATConv and GATv2Conv Layers #411

Closed achiverram28 closed 3 months ago

achiverram28 commented 3 months ago

Creating dropout functionality inside the GATConv and GATConv2 Layers

Addressing the issue https://github.com/CarloLucibello/GraphNeuralNetworks.jl/issues/258

CarloLucibello commented 3 months ago

Nice. A test should be added. Checking that a dropout keyword is accepted in the forward is enough.

achiverram28 commented 3 months ago

Is it fine now?

CarloLucibello commented 3 months ago

Tests are not passing.

achiverram28 commented 3 months ago

I guess tests should pass now

6EBA12CE-D437-4AB9-9FB3-4FE0BC088483 EE7CE163-D67E-462B-BC71-ADF9B5D79FFB