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 Layer #404

Closed achiverram28 closed 3 months ago

achiverram28 commented 4 months ago

Creating dropout functionality inside the GATConv Layer

Addressing the issue #258

achiverram28 commented 4 months ago

@CarloLucibello do let me know, if this finds right, then I will extend the same to GATv2Conv

Hoping that implementing dropout function, is automatically taken during trainmode! and testmode! as per discussions in the issues #1084 and in #1263

achiverram28 commented 3 months ago

Sorry ,for that I had forgot to rebase , I am closing this , as I am doing it from my new branch... Will take care of this properly!!