Skip to content

Commit

Permalink
Fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
sgugger authored and elusenji committed Jun 12, 2022
1 parent e522b82 commit 315d9e8
Showing 1 changed file with 16 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -288,8 +288,10 @@ def main():
"--data_file",
type=str,
default=None,
help="A jbl file containing tokenized data which can be split as objective dataset, "
"train_dataset and test_dataset.",
help=(
"A jbl file containing tokenized data which can be split as objective dataset, "
"train_dataset and test_dataset."
),
)

parser.add_argument(
Expand Down Expand Up @@ -319,8 +321,10 @@ def main():
"--context_len",
default=32,
type=int,
help="The maximum total input sequence length after tokenization. Sequences longer "
"than this will be truncated, sequences shorter will be padded.",
help=(
"The maximum total input sequence length after tokenization. Sequences longer "
"than this will be truncated, sequences shorter will be padded."
),
)

parser.add_argument(
Expand Down Expand Up @@ -350,8 +354,10 @@ def main():
"--eval_interval",
default=10,
type=int,
help="decay the selectivity of our secondary learner filter from"
"1 standard deviation above average to 1 below average after 10 batches",
help=(
"decay the selectivity of our secondary learner filter from"
"1 standard deviation above average to 1 below average after 10 batches"
),
)

parser.add_argument(
Expand All @@ -372,8 +378,10 @@ def main():
"--threshold",
default=1.0,
type=float,
help="The threshold value used by secondary learner to filter the train_data and allow only"
" informative data as input to the model",
help=(
"The threshold value used by secondary learner to filter the train_data and allow only"
" informative data as input to the model"
),
)

parser.add_argument("--finetuned_model_name", default="gpt2_finetuned.pt", type=str, help="finetuned_model_name")
Expand Down

0 comments on commit 315d9e8

Please sign in to comment.