cleaning up demos

This commit is contained in:
William Falcon
2019-10-05 16:35:39 -04:00
parent f1caeb49cc
commit 602236ecf2
+3 -3
View File
@@ -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',