File size: 397 Bytes
80d096b |
1 2 |
CUDA_VISIBLE_DEVICES=0 python softner_segmenter.py --model_name_or_path word_piece_seg/ --output_dir bert-word-piece-seg --data_dir ./ip_seg/ --model_type bert --labels ./labels_seg.txt --max_seq_length 128 --num_train_epochs 5 --learning_rate 5e-5 --per_gpu_train_batch_size 4 --per_gpu_train_batch_size 4 --save_steps 100000 --seed 1 --do_train --do_eval --do_predict --overwrite_output_dir |