main_train_model.py: Training file.
mask_lstm_model: VCSSM model file.