fix use monotonic time

learning-at-home · Jun 10, 2024 · c84e6dd · c84e6dd
1 parent 050bb63
commit c84e6dd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hivemind/averaging/averager.py b/hivemind/averaging/averager.py
@@ -701,7 +701,7 @@ async def _load_state_from_peers(self, future: MPFuture, timeout: Optional[float
             metadata = None
             for peer in sorted(peer_priority.keys(), key=peer_priority.get, reverse=True):
                 if peer != self.peer_id:
-                    t0 = time.time()
+                    t0 = time.monotonic()
                     logger.info(f"Downloading parameters from peer {peer}")
                     try:
                         stub = self.get_stub(self._p2p, peer, namespace=self.prefix)
@@ -724,7 +724,7 @@ async def _load_state_from_peers(self, future: MPFuture, timeout: Optional[float
                             logger.debug(f"Peer {peer} did not send its state")
                             continue
 
-                        t1 = time.time()
+                        t1 = time.monotonic()
                         logger.info(f"Finished downloading state in {t1 - t0:.3f}s from {peer}")
                         future.set_result((metadata, tensors))
                         return