https://github.com/google-research/xtreme/blob/master/scripts/train_xnli.sh