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 KL divergence between Deterministic + others for pytorch #36

Closed brendanhasz closed 3 years ago

brendanhasz commented 3 years ago

And refactor tests. Also added autoflake to dev stack.

Fixes https://github.com/brendanhasz/probflow/issues/35