Qihan Ren

Results 2 comments of Qihan Ren

I think it should be `criterion = metrics.ELBO(batch_size).to(device)` since the ELBO is calculated based on each mini-batch. Is it correct? @kumar-shridhar

Is the problem solved now? I come across the same problem with zero 3 on mix-modality training. The training always hangs at the make_experience stage, and the progress is always...