Add batched calculation option to `energy_score_empirical` in order to reduce memory consumption#3402

Merged

fritzo merged 2 commits intopyro-ppl:devfrom BenZickel:batched_energy_score_calculation

Sep 26, 2024

+67-6

Commits on Sep 24, 2024

Add batched calculation option to energy_score_empirical in order to reduce memory consumption.
Ben Zickel
committed

Commits on Sep 26, 2024

Replace native Python sum with torch stack(...).sum().
Ben Zickel
committed