Marigod98

maybe this part should be train_step < 50000? not should be epoch

Thanks for your wonderful work! I am interested in the model architecture: two branches, optimize post and prior distributions with KL_divergence and use some other loss to optimize decoder, I...