flo-schu / pymob

Python model building platform
MIT License
0 stars 0 forks source link

Implement early stopping in numpyro SVI #50

Open flo-schu opened 3 months ago

flo-schu commented 3 months ago

SVI tends to break, when the (local) optimum is reached. Why this happens, I don't know. It would be good to stop the loop, when the change in loss value becomes very small. This would a) speed up SVI and b) reduce non-converging chains. Also it allows greater control over SVI, for instance kicking out infinity simulations.