I have no problem running the example notebook, are you running on a different dataset? You can try to play around with the learning rate in obj_optimizer=pm.sgd(learning_rate=s).
I have no problem running the example notebook, are you running on a different dataset? You can try to play around with the learning rate in obj_optimizer=pm.sgd(learning_rate=s).