From 602236ecf2b5cec3b68b0a0a6d0feef8bc914c8b Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 16:35:39 -0400 Subject: [PATCH] cleaning up demos --- examples/basic_examples/gpu_template.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/basic_examples/gpu_template.py b/examples/basic_examples/gpu_template.py index 22ad6cbd..8c65f88b 100644 --- a/examples/basic_examples/gpu_template.py +++ b/examples/basic_examples/gpu_template.py @@ -51,9 +51,9 @@ if __name__ == '__main__': # gpu args parent_parser.add_argument( '--gpus', - type=str, - default='-1', - help='any integer (number of GPUs to use) or -1 for all' + type=int, + default=2, + help='how many gpus' ) parent_parser.add_argument( '--distributed_backend',