Hi,
I’m trying out different optimizers for VI using the obj_optimizer keyword like this:
advi_res = advi.fit(30000, obj_optimizer=pm.adam(), callbacks=[tracker])
I see here that it’s possible to add momentum to each optimizer but I can’t figure out the API for doing that. I’d appreciate some help.