first commit

6bc2da7 over 5 years ago

449 Bytes

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