Run with:
export WANDB_PROJECT=wangchanberta-qa
export MODEL_NAME=airesearchth/wangchanberta-base-wiki-20210520-spm
CUDA_LAUNCH_BLOCKING=1 python train_question_answering_lm_finetuning.py \
--model_name $MODEL_NAME \
--revision main@ckp-25000 \
--dataset_name iapp_thaiqa \
--output_dir $MODEL_NAME-finetune-iapp_thaiqa-model \
--log_dir $MODEL_NAME-finetune-iapp_thaiqa-log \
--model_max_length 400 \
--pad_on_right \
--fp16