Unverified Commit 9a5ef837 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[s2s] fix --gpus clarg collision (#6358)

parent 1aec9916
...@@ -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",
......
...@@ -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,
......
...@@ -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",
......
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