Method wait

python/singa/opt.py:822–824

Wait for the cuda streams used by the communicator to finish their operations.

(self)

                                                  topK)

    def wait(self):
        """Wait for the cuda streams used by the communicator to finish their operations."""
        self.communicator.wait()

    def backward_and_update(self, loss, threshold=2097152):
        """Performs backward propagation from the loss and parameter update.