pyro-ppl / numpyro / 13751371920

92%

Build Type

push

github

Committed by web-flow

Commit Message

Fix Memleak in ASVGD (#2003)

* added ASVGDState which tracks the annealing for ASVGD; added loss_temperature and repulsion_temperature to the SteinVIState; renamed _svgd_loss_and_grads => _loss_and_grads; added loss and repulsion temperature to signature of _loss_and_grads

* fixed minor issues; reproduces results from previous version

* added division by zero guard for ASVGD with num_cycles=1 and removed leak pytest.mark

* removed ASVGD memleak test from CI

Run Details

49 of 52 new or added lines in 1 file covered. (94.23%)

1 existing line in 1 file now uncovered.

13427 of 14463 relevant lines covered (92.84%)

1.99 hits per line