README.md

This model is developed with transformers v4.10.3.

Train

#!/usr/bin/env bash

export CUDA_VISIBLE_DEVICES=0

OUTDIR=bert-base-uncased-squad
WORKDIR=transformers/examples/pytorch/question-answering
cd $WORKDIR

nohup python run_qa.py \
    --model_name_or_path bert-base-uncased \
    --dataset_name squad \
    --do_eval \
    --do_train \
    --per_device_train_batch_size 16 \
    --per_device_eval_batch_size 16 \
    --doc_stride 128 \
    --max_seq_length 384 \
    --learning_rate 3e-5 \
    --num_train_epochs 2 \
    --eval_steps 250 \
    --save_steps 2500 \
    --logging_steps 1 \
    --overwrite_output_dir \
    --output_dir $OUTDIR 2>&1 | tee $OUTDIR/run.log &

Eval

export CUDA_VISIBLE_DEVICES=0

OUTDIR=eval-bert-base-uncased-squad
WORKDIR=transformers/examples/pytorch/question-answering
cd $WORKDIR

nohup python run_qa.py  \
    --model_name_or_path vuiseng9/bert-base-uncased-squad  \
    --dataset_name squad  \
    --do_eval  \
    --per_device_eval_batch_size 16  \
    --max_seq_length 384  \
    --doc_stride 128  \
    --overwrite_output_dir \
    --output_dir $OUTDIR 2>&1 | tee $OUTDIR/run.log &