Browse Source

Update hivemind/averaging/averager.py

Co-authored-by: Alexander Borzunov <hxrussia@gmail.com>
justheuristic 4 years ago
parent
commit
c24df06dcb
1 changed files with 1 additions and 1 deletions
  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.