-
Notifications
You must be signed in to change notification settings - Fork 1
/
run_NLG.sh
12 lines (9 loc) · 1.1 KB
/
run_NLG.sh
1
2
3
4
5
6
7
8
9
10
11
12
CUDA_VISIBLE_DEVICES=1 python main.py --model_type=gpt2 --model_name_or_path gpt2 --shots 5 --task RNNLG --length 50
CUDA_VISIBLE_DEVICES=1 python main.py --model_type=gpt2 --model_name_or_path gpt2 --shots 10 --task RNNLG --length 50
CUDA_VISIBLE_DEVICES=1 python main.py --model_type=gpt2 --model_name_or_path gpt2 --shots 20 --task RNNLG --length 50
CUDA_VISIBLE_DEVICES=1 python main.py --model_type=gpt2-large --model_name_or_path gpt2-large --shots 5 --task RNNLG --length 50
CUDA_VISIBLE_DEVICES=1 python main.py --model_type=gpt2-large --model_name_or_path gpt2-large --shots 10 --task RNNLG --length 50
CUDA_VISIBLE_DEVICES=1 python main.py --model_type=gpt2-large --model_name_or_path gpt2-large --shots 20 --task RNNLG --length 50
CUDA_VISIBLE_DEVICES=3 python main.py --model_type=gpt2-xl --model_name_or_path gpt2-xl --shots 5 --task RNNLG --length 50
CUDA_VISIBLE_DEVICES=3 python main.py --model_type=gpt2-xl --model_name_or_path gpt2-xl --shots 10 --task RNNLG --length 50
CUDA_VISIBLE_DEVICES=3 python main.py --model_type=gpt2-xl --model_name_or_path gpt2-xl --shots 20 --task RNNLG --length 50