Wasserstein-AutoEncoders icon indicating copy to clipboard operation
Wasserstein-AutoEncoders copied to clipboard

Possible Issue in MMD calculation

Open DaehanKim opened this issue 2 years ago • 0 comments

I think MMD approximation with batch size n should be dists_x divided by n(n-1) (analogous for dists_y) and dists_c divided by n**2. But this Implementation only divides with n-1 and n respectively. Why is it so?

https://github.com/schelotto/Wasserstein-AutoEncoders/blob/master/wae_mmd.py#L147-L149

DaehanKim avatar Oct 08 '22 15:10 DaehanKim