Re-balancing Variational Autoencoder Loss for Molecule Sequence Generation