Method __init__

sync_batchnorm/comm.py:69–78

Args: master_callback: a callback to be invoked after having collected messages from slave devices.

(self, master_callback)

    """

    def __init__(self, master_callback):
        """

        Args:
            master_callback: a callback to be invoked after having collected messages from slave devices.
        """
        self._master_callback = master_callback
        self._queue = queue.Queue()
        self._registry = collections.OrderedDict()
        self._activated = False

    def __getstate__(self):
        return {"master_callback": self._master_callback}