Added unit test @test_elbo_detailed_loss_sampler in variational_estimator_test.py

piEsposito / blitz-bayesian-deep-learning

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.

GNU General Public License v3.0

918 stars 107 forks source link

Added unit test @test_elbo_detailed_loss_sampler in variational_estimator_test.py #95

Closed filljonas closed 2 years ago

filljonas commented 2 years ago

Thanks for your feedback. I added a unit test for my new function in _variational_estimatortest.py.

Please let me know if there is anything else!

piEsposito commented 2 years ago

Hi @filljonas, you are the best. Thank you for your contributions, that saves me a lot of time and helps BLiTZ become better suited for different use cases. If you have any ideas or want to be more involved with the project let me know and we can talk.

As this is an independent project with very restricted resources, any help means a lot.

Also I'm merging it.

filljonas commented 2 years ago

Thank you for your helpful feedback! I will let you know if I have any ideas for the project in the future.