[s2s] fix --gpus clarg collision (#6358)
This commit is contained in:
@@ -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):
|
|||||||
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):
|
|||||||
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",
|
||||||
|
|||||||
Reference in New Issue
Block a user