Add dropout to neural estimators

cbg-ethz / bmi

Mutual information estimators and benchmark

https://cbg-ethz.github.io/bmi/

MIT License

33 stars 5 forks source link

Closed pawel-czyz closed 3 days ago

pawel-czyz commented 8 months ago

MINE, InfoNCE, and other neural estimators could use random state to use dropout. This requires changes in the training loop and some refactoring.