Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
9a5ef837
Unverified
Commit
9a5ef837
authored
Aug 08, 2020
by
Sam Shleifer
Committed by
GitHub
Aug 08, 2020
Browse files
[s2s] fix --gpus clarg collision (#6358)
parent
1aec9916
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
12 additions
and
6 deletions
+12
-6
examples/lightning_base.py
examples/lightning_base.py
+0
-5
examples/text-classification/run_pl_glue.py
examples/text-classification/run_pl_glue.py
+6
-1
examples/token-classification/run_pl_ner.py
examples/token-classification/run_pl_ner.py
+6
-0
No files found.
examples/lightning_base.py
View file @
9a5ef837
...
@@ -277,11 +277,6 @@ def add_generic_args(parser, root_dir) -> None:
...
@@ -277,11 +277,6 @@ def add_generic_args(parser, root_dir) -> None:
required
=
True
,
required
=
True
,
help
=
"The output directory where the model predictions and checkpoints will be written."
,
help
=
"The output directory where the model predictions and checkpoints will be written."
,
)
)
parser
.
add_argument
(
"--gpus"
,
default
=
0
,
type
=
int
,
help
=
"The number of GPUs allocated for this, it is by default 0 meaning none"
,
)
parser
.
add_argument
(
parser
.
add_argument
(
"--fp16"
,
"--fp16"
,
action
=
"store_true"
,
action
=
"store_true"
,
...
...
examples/text-classification/run_pl_glue.py
View file @
9a5ef837
...
@@ -155,7 +155,12 @@ class GLUETransformer(BaseTransformer):
...
@@ -155,7 +155,12 @@ class GLUETransformer(BaseTransformer):
parser
.
add_argument
(
parser
.
add_argument
(
"--task"
,
default
=
""
,
type
=
str
,
required
=
True
,
help
=
"The GLUE task to run"
,
"--task"
,
default
=
""
,
type
=
str
,
required
=
True
,
help
=
"The GLUE task to run"
,
)
)
parser
.
add_argument
(
"--gpus"
,
default
=
0
,
type
=
int
,
help
=
"The number of GPUs allocated for this, it is by default 0 meaning none"
,
)
parser
.
add_argument
(
parser
.
add_argument
(
"--data_dir"
,
"--data_dir"
,
default
=
None
,
default
=
None
,
...
...
examples/token-classification/run_pl_ner.py
View file @
9a5ef837
...
@@ -168,6 +168,12 @@ class NERTransformer(BaseTransformer):
...
@@ -168,6 +168,12 @@ class NERTransformer(BaseTransformer):
type
=
str
,
type
=
str
,
help
=
"Path to a file containing all labels. If not specified, CoNLL-2003 labels are used."
,
help
=
"Path to a file containing all labels. If not specified, CoNLL-2003 labels are used."
,
)
)
parser
.
add_argument
(
"--gpus"
,
default
=
0
,
type
=
int
,
help
=
"The number of GPUs allocated for this, it is by default 0 meaning none"
,
)
parser
.
add_argument
(
parser
.
add_argument
(
"--data_dir"
,
"--data_dir"
,
...
...
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