github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix type of num_worker_threads (#2237) 

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
pull/2239/head^2
Ziyue Jiang 2022-12-30 11:04:01 +08:00 committed by GitHub
parent db4cbdc7fb
commit 57929a6210
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/pipeline/rpc/utils.py
View File

@ -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()