Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
47107028
Commit
47107028
authored
May 18, 2022
by
Sylvain Gugger
Browse files
Fix style
parent
5fdb54ec
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
16 additions
and
8 deletions
+16
-8
examples/research_projects/information-gain-filtration/run_clm_igf.py
...earch_projects/information-gain-filtration/run_clm_igf.py
+16
-8
No files found.
examples/research_projects/information-gain-filtration/run_clm_igf.py
View file @
47107028
...
@@ -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"
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment