search

bayesmech / research-gnn-pooling

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

Stochastic Diff-Pooling Method #6

Closed AnimeshSinha1309 closed 2 years ago

AnimeshSinha1309 commented 2 years ago

This module implements a memory-efficient method of diff-pool-like pooling when the pooling amount is not abrupt.

Closes #7