yaringal / ConcreteDropout

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

ConcreteDropout

Code for Concrete Dropout as presented in https://arxiv.org/abs/1705.07832

Update (08/2018)

Added SpatialConcreteDropout for use with convolution layers. Please don't use ConcreteDropout with conv layers as the regulariser weighing is wrong for convolutions.