From b261e4b80f48fc0cfb6bcd2ff140b67b0cdb82bf Mon Sep 17 00:00:00 2001 From: Yuchen Zhong Date: Sat, 9 May 2020 21:11:27 +0800 Subject: [PATCH] hotfix: use default num_threads (#7) --- byteps/mxnet/compression.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/byteps/mxnet/compression.py b/byteps/mxnet/compression.py index d5cecb9e2..fd90f2cb3 100644 --- a/byteps/mxnet/compression.py +++ b/byteps/mxnet/compression.py @@ -66,7 +66,7 @@ def decompress(self, tensor, ctx, *args, **kwargs): class WeightDecayMomentum(Compressor): """For 1bit compression.""" - pool = concurrent.futures.ThreadPoolExecutor(max_workers=4) + pool = concurrent.futures.ThreadPoolExecutor() def __init__(self, compressor, mu, wd): self.compressor = compressor