Commit 47107028 authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Fix style

parent 5fdb54ec
...@@ -288,8 +288,10 @@ def main(): ...@@ -288,8 +288,10 @@ def main():
"--data_file", "--data_file",
type=str, type=str,
default=None, default=None,
help="A jbl file containing tokenized data which can be split as objective dataset, " help=(
"train_dataset and test_dataset.", "A jbl file containing tokenized data which can be split as objective dataset, "
"train_dataset and test_dataset."
),
) )
parser.add_argument( parser.add_argument(
...@@ -319,8 +321,10 @@ def main(): ...@@ -319,8 +321,10 @@ def main():
"--context_len", "--context_len",
default=32, default=32,
type=int, type=int,
help="The maximum total input sequence length after tokenization. Sequences longer " help=(
"than this will be truncated, sequences shorter will be padded.", "The maximum total input sequence length after tokenization. Sequences longer "
"than this will be truncated, sequences shorter will be padded."
),
) )
parser.add_argument( parser.add_argument(
...@@ -350,8 +354,10 @@ def main(): ...@@ -350,8 +354,10 @@ def main():
"--eval_interval", "--eval_interval",
default=10, default=10,
type=int, type=int,
help="decay the selectivity of our secondary learner filter from" help=(
"1 standard deviation above average to 1 below average after 10 batches", "decay the selectivity of our secondary learner filter from"
"1 standard deviation above average to 1 below average after 10 batches"
),
) )
parser.add_argument( parser.add_argument(
...@@ -372,8 +378,10 @@ def main(): ...@@ -372,8 +378,10 @@ def main():
"--threshold", "--threshold",
default=1.0, default=1.0,
type=float, type=float,
help="The threshold value used by secondary learner to filter the train_data and allow only" help=(
" informative data as input to the model", "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") parser.add_argument("--finetuned_model_name", default="gpt2_finetuned.pt", type=str, help="finetuned_model_name")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment