yaringal / ConcreteDropout

Code for Concrete Dropout as presented in https://arxiv.org/abs/1705.07832
MIT License
246 stars 68 forks source link

Added a Tensorflow version #2

Closed joeyearsley closed 7 years ago

joeyearsley commented 7 years ago

Re-implemented in almost pure TF following the implementation style of tf.layers.

Almost pure as I needed to use tf.contrib.keras's Wrapper layer.

yaringal commented 7 years ago

cool, thanks!