Update PyTorch example - Githubissues

yaringal / ConcreteDropout

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

MIT License

246 stars 68 forks source link

Update PyTorch example #9

Closed y0ast closed 5 years ago

y0ast commented 5 years ago

Cleans up the PyTorch code (now requires 0.4+)
Fixes bug with using a ReLU on output layer
Fixes bug with summing over batch dimension in heteroscedastic loss

Results are same/similar.

Look here for a GitHub rendered version of the new notebook: https://github.com/yaringal/ConcreteDropout/blob/622e6a12a99005cf660f266e6d2d2b69ccb688ce/concrete-dropout-pytorch.ipynb