bayesmech / research-gnn-pooling

Research on Heirarchical pooling for Graph Neural Networks
MIT License
0 stars 0 forks source link

Visualize Gradients on WandB #10

Open AnimeshSinha1309 opened 2 years ago

AnimeshSinha1309 commented 2 years ago

Log the gradients to WandB, visualize and understand how the stochastic samples compare to the deterministic form.