13 lines
337 B
Bash
13 lines
337 B
Bash
|
python train.py \
|
||
|
|
--do_eval \
|
||
|
|
--debug True \
|
||
|
|
--device gpu \
|
||
|
|
--model_name_or_path checkpoint \
|
||
|
|
--output_dir checkpoint \
|
||
|
|
--per_device_eval_batch_size 32 \
|
||
|
|
--max_length 256 \
|
||
|
|
--label_path ./data/label.txt \
|
||
|
|
--train_path ./data/train.txt \
|
||
|
|
--test_path ./data/test.txt \
|
||
|
|
--dev_path ./data/dev.txt
|