Skip to content

Commit

Permalink
bugfix(args): fixed conflicting option string
Browse files Browse the repository at this point in the history
  • Loading branch information
DongjunLee committed Sep 10, 2019
1 parent b728cb9 commit 7898b1c
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions claf/config/args.py
Original file line number Diff line number Diff line change
Expand Up @@ -1098,15 +1098,15 @@ def model(parser):

group = parser.add_argument_group(f" # RoBERTa")
group.add_argument(
"--roberta_for_seq_cls.pretrained_model_name",
type=str, default=None, dest="model.roberta_for_seq_cls.pretrained_model_name",
"--roberta_for_tok_cls.pretrained_model_name",
type=str, default=None, dest="model.roberta_for_tok_cls.pretrained_model_name",
help=""" A str with the name of a pre-trained model to load selected in the list of (default: None):
. `roberta-base`
. `roberta-large` """,
)
group.add_argument(
"--roberta_for_seq_cls.dropout",
type=float, default=0.2, dest="model.roberta_for_seq_cls.dropout",
"--roberta_for_tok_cls.dropout",
type=float, default=0.2, dest="model.roberta_for_tok_cls.dropout",
help=""" The prob of fc layer dropout """
)

Expand Down

0 comments on commit 7898b1c

Please sign in to comment.