From 225b3fda9191a91da8f7f8e701bfd676d927a348 Mon Sep 17 00:00:00 2001
From: Howard Lau <howardlau1999@hotmail.com>
Date: Tue, 25 May 2021 03:05:33 +0800
Subject: [PATCH] [tune] Fix ddp_mnist_torch example argument type (#16017)

* fix ddp_mnist_torch example argument type

* add period
---
 python/ray/tune/examples/ddp_mnist_torch.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/ray/tune/examples/ddp_mnist_torch.py b/python/ray/tune/examples/ddp_mnist_torch.py
index 9c71fb612..4430fe7bc 100644
--- a/python/ray/tune/examples/ddp_mnist_torch.py
+++ b/python/ray/tune/examples/ddp_mnist_torch.py
@@ -71,9 +71,9 @@ if __name__ == "__main__":
         help="Sets number of workers for training.")
     parser.add_argument(
         "--num-gpus-per-worker",
-        action="store_true",
-        default=False,
-        help="enables CUDA training")
+        type=int,
+        default=0,
+        help="Sets number of gpus each worker uses.")
     parser.add_argument(
         "--cluster",
         action="store_true",