Procházet zdrojové kódy

remove debug messages

justheuristic před 4 roky
rodič
revize
1528c8c050
2 změnil soubory, kde provedl 2 přidání a 4 odebrání
  1. 1 2
      hivemind/averaging/averager.py
  2. 1 2
      hivemind/dht/__init__.py

+ 1 - 2
hivemind/averaging/averager.py

@@ -120,7 +120,7 @@ class DecentralizedAverager(mp.Process, ServicerBase):
         assert all(bit in "01" for bit in initial_group_bits)
         assert not client_mode or not auxiliary, "auxiliary peers must accept incoming connections"
 
-        super().__init__(name='DecentralizedAverager')
+        super().__init__()
         self.dht = dht
         self.prefix = prefix
 
@@ -207,7 +207,6 @@ class DecentralizedAverager(mp.Process, ServicerBase):
         Turns out, using a non-main thread creates a separate OMP pool that works even if the original pool is corrupted
         Read more: https://github.com/pytorch/pytorch/issues/17199
         """
-        print(f"RUNNING AVERAGER WITH PID={self.pid}")
         thread = threading.Thread(target=self._run_internal, daemon=True)
         thread.start()
         thread.join()

+ 1 - 2
hivemind/dht/__init__.py

@@ -70,7 +70,7 @@ class DHT(mp.Process):
         **kwargs,
     ):
         self._parent_pid = os.getpid()
-        super().__init__(name="DHT")
+        super().__init__()
 
         if not (
             initial_peers is None
@@ -102,7 +102,6 @@ class DHT(mp.Process):
 
     def run(self) -> None:
         """Serve DHT forever. This function will not return until DHT node is shut down"""
-        print(f"RUNNING DHT WITH PID={self.pid}")
 
         loop = switch_to_uvloop()
         pipe_semaphore = asyncio.Semaphore(value=0)