浏览代码

Update hivemind/averaging/averager.py

Co-authored-by: Alexander Borzunov <hxrussia@gmail.com>
justheuristic 4 年之前
父节点
当前提交
c24df06dcb
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      hivemind/averaging/averager.py

+ 1 - 1
hivemind/averaging/averager.py

@@ -555,7 +555,7 @@ class DecentralizedAverager(mp.Process, ServicerBase):
         return await future
 
     def load_state_from_peers(
-        self, wait=True, timeout: Optional[float] = None
+        self, wait: bool = True, timeout: Optional[float] = None
     ) -> Optional[Tuple[Any, Sequence[torch.Tensor]]]:
         """
         Try to download the latest optimizer state one of the existing peer.