fix type of num_worker_threads (#2237)

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
pull/2239/head^2
Ziyue Jiang 2 years ago committed by GitHub
parent db4cbdc7fb
commit 57929a6210
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save