search

JavierAntoran / Bayesian-Neural-Networks

Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more
MIT License
1.83k stars 302 forks source link

Update optimizers.py #26

Closed lan-qing closed 1 year ago

lan-qing commented 1 year ago

replace weight_decay as self.weight_decay to fix NameError