dist.Bernoulli recognize the right batch_size, but dist.Categorical won't

[BigDiaos]

Sure, I’ll never mind! See this topic Is it possible to build reconstruction loss to learn sequential/positional relations among data entries? - Tutorials - Pyro Discussion Forum

Thank for your help!