source sft_venv/bin/activate
cd autodl-tmp/
python sft_mini_bert.py \
  --pretrained_model_path /root/autodl-tmp/mini-bert-output/step-20000 \
  --tokenizer_path /root/autodl-tmp/mini-bert-output/tokenizer \
  --train_data sst2_train.csv \
  --val_data sst2_validation.csv \
  --output_dir /root/autodl-tmp/mini-bert-sft-output \
  --num_epochs 3 \
  --batch_size 4 \
  --learning_rate 2e-5 \
  --max_length 64