-
Notifications
You must be signed in to change notification settings - Fork 9
/
train.sh
27 lines (18 loc) · 2.39 KB
/
train.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#(c) 2021 NCSOFT Corporation & Korea University. All rights reserved.
#Train the model only with LM
#CUDA_VISIBLE_DEVICES=4 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --flag E2_L10_LM --ps_coef 0 --kn_coef 0 --model_name GPT2 --incontext > train_log_focus/e2_L20_LM_gpt.log &
#CUDA_VISIBLE_DEVICES=5 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --flag E2_L10_LM --ps_coef 0 --kn_coef 0 --model_name BART --incontext > train_log_focus/e2_L20_LM_bart.log &
#CUDA_VISIBLE_DEVICES=6 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --flag E2_L10_LM --ps_coef 0 --kn_coef 0 --model_name transformer-decoder --incontext > train_log_focus/e2_L20_LM_decoder.log &
#CUDA_VISIBLE_DEVICES=7 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --flag E2_L10_LM --ps_coef 0 --kn_coef 0 --model_name transformer-encdec --incontext > train_log_focus/e2_L20_LM_encdec.log &
#Train the model with Knowledge Selection, Persona Selection and Language Modeling
CUDA_VISIBLE_DEVICES=1 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --flag E2_L10 --model_name GPT2 --incontext > train_log_focus/e2_L20_gpt.log &
#CUDA_VISIBLE_DEVICES=5 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --flag E2_L10 --model_name BART --incontext > train_log_focus/e2_L20_bart.log &
#CUDA_VISIBLE_DEVICES=6 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --flag E2_L10 --model_name transformer-decoder --incontext > train_log_focus/e2_L20_decoder.log &
#CUDA_VISIBLE_DEVICES=7 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --flag E2_L10 --model_name transformer-encdec --incontext > train_log_focus/e2_L20_encdec.log &
#Train the model w/o Knowledge Selection
#CUDA_VISIBLE_DEVICES=0 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --kn_coef 0 --flag E2_L10_WO_KS --model_name BART --incontext > train_log_focus/e2_L20_bart_WO_KS.log &
#CUDA_VISIBLE_DEVICES=1 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --kn_coef 0 --flag E2_L10_WO_KS --model_name GPT2 --incontext > train_log_focus/e2_L20_gpt_WO_KS.log &
#Train the model w/o Persona Selection
#CUDA_VISIBLE_DEVICES=0 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --ps_coef 0 --flag E2_L10_WO_PS --model_name GPT2 --incontext > train_log_focus/e2_L20_gpt_WO_PS.log &
#CUDA_VISIBLE_DEVICES=1 nohup python train_focus.py --n_epochs 2 --lm_coef 10 --ps_coef 0 --flag E2_L10_WO_PS --model_name BART --incontext > train_log_focus/e2_L20_bart_WO_PS.log &
echo