From 57929a6210bc822d87d45d49063047cf01d2d9a6 Mon Sep 17 00:00:00 2001 From: Ziyue Jiang Date: Fri, 30 Dec 2022 11:04:01 +0800 Subject: [PATCH] fix type of num_worker_threads (#2237) Co-authored-by: Ziyue Jiang --- colossalai/pipeline/rpc/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colossalai/pipeline/rpc/utils.py b/colossalai/pipeline/rpc/utils.py index 4310b3afe..06e6d976d 100644 --- a/colossalai/pipeline/rpc/utils.py +++ b/colossalai/pipeline/rpc/utils.py @@ -151,5 +151,5 @@ def parse_args(): parser.add_argument('--device', type=str, choices=['cpu', 'cuda'], default='cuda') parser.add_argument('--master_addr', type=str, default='localhost') parser.add_argument('--master_port', type=str, default='29020') - parser.add_argument('--num_worker_threads', type=str, default=128) + parser.add_argument('--num_worker_threads', type=int, default=128) return parser.parse_args()