From e34113625a1629e10268a4dfeb16ca6287710c71 Mon Sep 17 00:00:00 2001 From: Kevin Meng Date: Tue, 6 Jul 2021 11:54:20 -0400 Subject: [PATCH] use adversarial training --- exec_transf_exp.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/exec_transf_exp.sh b/exec_transf_exp.sh index 6adacfc..6374ac8 100644 --- a/exec_transf_exp.sh +++ b/exec_transf_exp.sh @@ -2,5 +2,5 @@ for i in {1..3} do # CUDA_VISIBLE_DEVICES=$i nohup sh -c "python3 -m adv_transformer.train --cs_model_dir=output_bert_$i --cs_tfm_type=bert-base-uncased --cs_train_steps=10 --cs_perturb_id=6 --cs_adv_train=True"&>nohup_bert_$i.out& # CUDA_VISIBLE_DEVICES=$i nohup sh -c "python3 -m adv_transformer.train --cs_model_dir=output_distilbert_$i --cs_tfm_type=distilbert-base-uncased --cs_train_steps=7 --cs_adv_train=True"&>nohup_distilbert_$i.out& -CUDA_VISIBLE_DEVICES=$i nohup sh -c "python3 -m adv_transformer.train --cs_model_dir=output_roberta_$i --cs_tfm_type=roberta-base --cs_train_steps=30 --cs_tfm_ft_enc_layers=6"&>nohup_roberta_$i.out& +CUDA_VISIBLE_DEVICES=$i nohup sh -c "python3 -m adv_transformer.train --cs_model_dir=output_roberta_$i --cs_tfm_type=roberta-base --cs_train_steps=30 --cs_tfm_ft_enc_layers=6 --cs_adv_train=True"&>nohup_roberta_$i.out& done \ No newline at end of file