hotfix: use concurrent.futures.Timeout (#8)

* hotfix: fix timeout typo * hotfix: test timeout * hotfix: set timeout 0.1s
bytedance · Jun 23, 2020 · f6fd608 · f6fd608
1 parent b261e4b
commit f6fd608
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/byteps/mxnet/__init__.py b/byteps/mxnet/__init__.py
@@ -230,9 +230,6 @@ def __init__(self, params, optimizer, optimizer_params=None, root_rank=0, compre
                 )
                 byteps_declare_tensor("gradient_" + str(i), **byteps_params)
 
-    def __del__(self):
-        self._f.close()
-
     def _register_compressor(self, params, optimizer_params, compression_params):
         """Register compressor for BytePS
 

diff --git a/byteps/mxnet/compression.py b/byteps/mxnet/compression.py
@@ -105,8 +105,8 @@ def decompress(self, tensor, ctx, *args, **kwargs):
         try:
             self.future.result(timeout=0.1)
             tensor += self.cache
-        except TimeoutError:
-            print("timeout")
+        except concurrent.futures.TimeoutError:
+            print("timeout for wd-momentum")
         return self.compressor.decompress(tensor, ctx)