You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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.The text was updated successfully, but these errors were encountered: