Преглед на файлове

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.