search

thu-ml / zhusuan

A probabilistic programming library for Bayesian deep learning, generative models, based on Tensorflow
http://zhusuan.readthedocs.io
MIT License
2.21k stars 418 forks source link

Eager executation #122

Open rzu512 opened 5 years ago

rzu512 commented 5 years ago

When I try to scale up my model, I find that tf.scan uses too much memory. Probably I have to use eager execution.

How well does HMC and SGMCMC work with eager execution and tensorflow 2.0?

thjashin commented 5 years ago

That's in our plan but need quite amount of work to switch to the new-style of variable keeping mechanism.

thjashin commented 5 years ago

Welcome for contribution!