hotfix: use default num_threads (#7)

bytedance · Jun 23, 2020 · b261e4b · b261e4b
1 parent 7a0bea6
commit b261e4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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