Test the idea of Regularization by sampling

bayesmech / research-gnn-pooling

Research on Heirarchical pooling for Graph Neural Networks

MIT License

0 stars 0 forks source link

Test the idea of Regularization by sampling #13

Open AnimeshSinha1309 opened 2 years ago

AnimeshSinha1309 commented 2 years ago

Replace Gumbel Softmax with Softmax with straight-through. This will help us check if it's like Edge dropout which is helping us improve on validation even when we lose on train.