def METHOD_NAME(self, closure=None):
    """Performs a single optimization step."""
    self._optimizer.METHOD_NAME(closure)
    self.set_num_updates(self.get_num_updates() + 1)
    if self._is_warmup_end():
        self._warmup_sync()
    elif self._is_bmuf_iter():
        self._block_sync()