There is an RV named phi
with dim=5000
in my model. When I tried to build a guide using AutoNormal
or AutoLowRankMultivariateNormal
. I got the following error:
Error while computing log_prob at site 'phi_songId':
Expected value argument (Tensor of shape (6, 5000)) to be within the support (Simplex()) of the distribution Dirichlet(concentration: torch.Size([6, 5000])), but found invalid values:
tensor([[0.0002, 0.0002, 0.0002, ..., 0.0002, 0.0002, 0.0002],
[0.0002, 0.0002, 0.0002, ..., 0.0002, 0.0002, 0.0002],
[0.0002, 0.0002, 0.0002, ..., 0.0002, 0.0002, 0.0002],
[0.0002, 0.0002, 0.0002, ..., 0.0002, 0.0002, 0.0002],
[0.0002, 0.0002, 0.0002, ..., 0.0002, 0.0002, 0.0002],
[0.0002, 0.0002, 0.0002, ..., 0.0002, 0.0002, 0.0002]],
device='cuda:1', grad_fn=<ExpandBackward0>)
Trace Shapes:
Param Sites:
Sample Sites:
phi_songId dist 6 | 5000
value 6 | 5000
What can I do to solve this problem?