Great package. Finally a nice PPL for PyTorch! Thanks!
My question is if it is enough for me to do this (following the Neural HMM code):
pyro.sample("obs_x_%d" % t,
log_pdf_mask=mini_batch_mask[:, t - 1:t],
obs=mini_batch[:, t - 1].unsqueeze(1),
or should I provide
vs=something as well, as my observations are just indices of the classes, instead of one hots.