run_cls.sh 384 字节
Newer Older
Z
zhanghan17 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
source $1

python3 -m paddle.distributed.launch ./ernie_gram/finetune_classifier_distributed.py \
    --data_dir $data_dir \
    --max_steps $max_steps \
    --bsz $bsz \
    --lr $lr \
    --label_map ${label_map:-""} \
    --num_labels $num_labels \
    --pair_input $pair_input \
    --valid_steps $valid_steps \
    --from_pretrained $from_pretrained \
    --save_dir checkpoints