Has anyone had success training the unconditional generation model? Any checkpoints or idea of how many epochs it takes to converge?