probaforms icon indicating copy to clipboard operation
probaforms copied to clipboard

Allow randomness='same' with RealNVP.samples

Open hombit opened this issue 1 year ago • 0 comments

We can use NFs not for variance estimation only, but for covariance estimation as well. For this we need to sample input batch with the same random seeds. I've found that it is probably achievable with torch.func.vmap(randomness='same') stabilized in PyTorch v2 (see #7 for torch upgrading issue), but I cannot make it work.

hombit avatar Jul 10 '23 20:07 hombit