brendanhasz / probflow

A Python package for building Bayesian models with TensorFlow or PyTorch
http://probflow.readthedocs.io
MIT License
171 stars 17 forks source link

Add batch_norm kwarg to DenseNetwork #38

Closed brendanhasz closed 3 years ago

brendanhasz commented 3 years ago

Add batch_norm kwarg to DenseNetwork (a bool).

Also additional kwarg, batch_norm_loc either after (adds batch norm after the activation function, the default) or before (adds batch norm before the activation func, as originally suggested).