Skip to content

GitLab

Unverified Commit 57929a62 authored by Ziyue Jiang's avatar Ziyue Jiang Committed by GitHub
Browse files

fix type of num_worker_threads (#2237) 


Co-authored-by: default avatarZiyue Jiang <ziyue.jiang@gmail.com>
parent db4cbdc7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/pipeline/rpc/utils.py
View file @ 57929a62
......@@ -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()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment