To train the models, we use the run_train.py CLI script. The script supports various arguments. See the training/README.md file for more information, along with the hyperparameters used in the paper.