How to use run_classification.py to fine tuning bert in imdb?
dataset="imdb "
python run_classification.py \
    --model_name_or_path  bert-base-uncased \
    --dataset_name imdb \
    --metric_name accuracy \
    --do_train \
    --max_seq_length 128 \
    --per_device_train_batch_size 32 \
    --learning_rate 2e-5 \
    --num_train_epochs 2 \
    --overwrite_output_dir \
    --output_dir tmp/imdb/When I run this command, it just ends, with no outputSnipaste_2023-10-01_20-30-091227×664 66.7 KB