Explorar el Código

Update hivemind/averaging/averager.py

Co-authored-by: Alexander Borzunov <hxrussia@gmail.com>
justheuristic hace 4 años
padre
commit
c24df06dcb
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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.