github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-18 19:58:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix type of num_worker_threads (#2237)

Browse Source 
Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
This commit is contained in:
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()