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

2020-08-08 21:51:37 -04:00
parent 1aec991643
commit 9a5ef83748
3 changed files with 12 additions and 6 deletions
--- a/examples/token-classification/run_pl_ner.py
+++ b/examples/token-classification/run_pl_ner.py
@@ -168,6 +168,12 @@ class NERTransformer(BaseTransformer):
            type=str,
            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(
            "--data_dir",