-
Notifications
You must be signed in to change notification settings - Fork 4
/
run.sh
19 lines (14 loc) · 1.97 KB
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#!/usr/bin/env bash
export DATA_DIR='./data/'
export MODEL_DIR='./pretrain_model/'
export OUTPUT_DIR='./output/lap'
#lap14
CUDA_VISIBLE_DEVICES=1 python ./examples/run_aclt.py --task_name lap --model_type bert --do_train --do_eval --do_test --do_lower_case --overwrite_output_dir --data_dir $DATA_DIR --model_name_or_path $MODEL_DIR --learning_rate 5e-5 --num_train_epochs 19 --max_seq_length 96 --per_gpu_train_batch_size 64 --per_gpu_eval_batch_size 64 --gamma 0.1 --output_dir $OUTPUT_DIR
#rest14
# CUDA_VISIBLE_DEVICES=1 python ./examples/run_aclt.py --task_name rest --model_type bert --do_train --do_eval --do_test --do_lower_case --overwrite_output_dir --data_dir $DATA_DIR --model_name_or_path $MODEL_DIR --learning_rate 5e-5 --num_train_epochs 9 --max_seq_length 96 --per_gpu_train_batch_size 64 --per_gpu_eval_batch_size 64 --gamma 0.1 --output_dir $OUTPUT_DIR
#rest15
# CUDA_VISIBLE_DEVICES=1 python ./examples/run_aclt.py --task_name rest15 --model_type bert --do_train --do_eval --do_test --do_lower_case --overwrite_output_dir --data_dir $DATA_DIR --model_name_or_path $MODEL_DIR --learning_rate 5e-5 --num_train_epochs 19 --max_seq_length 96 --per_gpu_train_batch_size 64 --per_gpu_eval_batch_size 64 --gamma 0.1 --output_dir $OUTPUT_DIR
#rest16
# CUDA_VISIBLE_DEVICES=1 python ./examples/run_aclt.py --task_name rest16 --model_type bert --do_train --do_eval --do_test --do_lower_case --overwrite_output_dir --data_dir $DATA_DIR --model_name_or_path $MODEL_DIR --learning_rate 5e-5 --num_train_epochs 26 --max_seq_length 96 --per_gpu_train_batch_size 64 --per_gpu_eval_batch_size 64 --gamma 0.1 --output_dir $OUTPUT_DIR
#twitter
#CUDA_VISIBLE_DEVICES=1 python ./examples/run_aclt.py --task_name twi --model_type bert --do_train --do_eval --do_test --do_lower_case --overwrite_output_dir --data_dir $DATA_DIR --model_name_or_path $MODEL_DIR --learning_rate 5e-5 --num_train_epochs 2 --max_seq_length 96 --per_gpu_train_batch_size 64 --per_gpu_eval_batch_size 64 --gamma 0.1 --output_dir $OUTPUT_DIR