Make bnn_mnist_sgld_blackjax.ipynb to be able to run in colab

probml / pyprobml

Python code for "Probabilistic Machine learning" book by Kevin Murphy

MIT License

6.46k stars 1.52k forks source link

Open karm-patel opened 2 years ago

karm-patel commented 2 years ago

currently, this notebook won't run in colab, due to requirement of high memory.
Mostly prediction loops taking much memory, we can make it batch prediction to consume less memory at a time