hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-08 19:41:38 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[tune] Fix ddp_mnist_torch example argument type (#16017)

Browse source 
* fix ddp_mnist_torch example argument type

* add period
This commit is contained in:
Howard Lau 2021-05-25 03:05:33 +08:00 • committed by GitHub
parent 0fecfe10b8
commit 225b3fda91
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/ray/tune/examples/ddp_mnist_torch.py
View file

@ -71,9 +71,9 @@ if __name__ == "__main__":
help="Sets number of workers for training.") help="Sets number of workers for training.")
parser.add_argument( parser.add_argument(
"--num-gpus-per-worker", "--num-gpus-per-worker",
action="store_true", type=int,
default=False, default=0,
help="enables CUDA training") help="Sets number of gpus each worker uses.")
parser.add_argument( parser.add_argument(
"--cluster", "--cluster",
action="store_true", action="store_true",