|
@@ -203,6 +203,7 @@ 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()
|