Before revision:  Next, we train a beta-VAE model, combining the GNN encoder from the first stage with an LSTM-based decoder with a customized loss.
After revision:  Next, we train a \beta-VAE model, combining the GNN encoder from the first stage with an LSTM-based decoder with a customized loss.