Jelajahi Sumber

Update hivemind/averaging/averager.py

Co-authored-by: Alexander Borzunov <hxrussia@gmail.com>
justheuristic 4 tahun lalu
induk
melakukan
c24df06dcb
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  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.