[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/text-classification/run_pl_glue.py
+++ b/examples/text-classification/run_pl_glue.py
@@ -155,7 +155,12 @@ class GLUETransformer(BaseTransformer):
        parser.add_argument(
            "--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(
            "--data_dir",
            default=None,