@@ -13,7 +13,7 @@ def add_args(parser):
1313 parser .add_argument ("--sub_task" , type = str , default = '' )
1414 parser .add_argument ("--lang" , type = str , default = '' )
1515 parser .add_argument ("--eval_task" , type = str , default = '' )
16- parser .add_argument ("--model_type" , default = "roberta " , type = str , choices = ['roberta' , 't5 ' , 'bart' , 'codet5' ])
16+ parser .add_argument ("--model_type" , default = "codet5 " , type = str , choices = ['roberta' , 'bart' , 'codet5' ])
1717 parser .add_argument ("--add_lang_ids" , action = 'store_true' )
1818 parser .add_argument ("--data_num" , default = - 1 , type = int )
1919 parser .add_argument ("--start_epoch" , default = 0 , type = int )
@@ -98,7 +98,7 @@ def add_args(parser):
9898 parser .add_argument ("--local_rank" , type = int , default = - 1 ,
9999 help = "For distributed training: local_rank" )
100100 parser .add_argument ('--seed' , type = int , default = 1234 ,
101- help = "random seed for initialization" ) # previous one 42
101+ help = "random seed for initialization" )
102102 args = parser .parse_args ()
103103
104104 if args .task in ['summarize' ]:
0 commit comments