CURRENT_DIR=`pwd`
export MODEL_DIR=$CURRENT_DIR/pretrained_models/bert-base-cn
export DATA_DIR=$CURRENT_DIR/dataset
export OUTPUR_DIR=$CURRENT_DIR/outputs
export TASK_NAME=fewshot
export MODEL_TYPE=bert

#-----------training-----------------
python examples/task_text_classify_fewshot_pet.py \
  --model_type=$MODEL_TYPE \
  --pretrained_model_path=$MODEL_DIR \
  --task_name=$TASK_NAME \
  --do_train \
  --do_lower_case \
  --device_id='0' \
  --checkpoint_mode=max \
  --experiment_name='ver_001' \
  --checkpoint_monitor=eval_acc \
  --data_dir=$DATA_DIR/$TASK_NAME/ \
  --train_input_file=fewshot_train.tsv \
  --eval_input_file=fewshot_dev.tsv \
  --output_dir=$OUTPUR_DIR/${TASK_NAME}_output/ \
  --train_max_seq_length=256 \
  --eval_max_seq_length=256 \
  --per_gpu_train_batch_size=8 \
  --per_gpu_eval_batch_size=16 \
  --learning_rate=3e-5 \
  --num_train_epochs=10 \
  --gradient_accumulation_steps=1 \
  --warmup_rate=0.1 \
  --scheduler_type=cosine \
  --scheduler_on=batch \
  --logging_steps=-1 \
  --save_steps=-1 \
  --seed=42