pip install transformers==4.5.1

export task=QYTM
export model_name=RE2
CUDA_VISIBLE_DEVICES=1 python src/DL_model/${model_name}/main.py \
  --task ${task} \
  --model_name ${model_name} \
  --output_dir ./output/${task}_${model_name}_20211007 \
  --tensorboardx_path  ./output/logs/runs/${task}_${model_name}_20211007  \
  --train_path ./datasets/${task}/train.csv \
  --dev_path ./datasets/${task}/dev.csv \
  --test_path ./datasets/${task}/test.csv \
  --label_file_level_dir ./datasets/${task}/labels_level.txt \
  --label2freq_level_dir ./datasets/${task}/label2freq_level.json \
  --vocab_file ./resources/word2vec/vocab.txt \
  --w2v_file ./resources/word2vec/token_vec_300.bin \
  --train_batch_size 256 \
  --eval_batch_siz 256 \
  --do_train \
  --do_eval \
  --do_predict \
  --num_train_epochs 50