Skip to content

Commit 88cf801

Browse files
authored
Update configs.py
1 parent 84687a2 commit 88cf801

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

configs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)