4cf9cfaa3ab7c77bf45f3e36079af7bc348515e8,python/src/nnabla/utils/data_iterator.py,DataIterator,next,#DataIterator#,174

Before Change

        """
        data = [[] for x in self._variables]
        batch_size = self._batch_size
        for b in range(batch_size):
            d = self._get_next_data()
            for i, v in enumerate(self._variables):
                data[i].append(d[i])
        return tuple([numpy.array(x) for x in data])

    def _callback_epoch_end(self):
        for callback in self._epoch_end_callbacks:

After Change

        """
        self._next_thread.join()
        data = self._current_data
        self._next_thread = threading.Thread(target=self._next)
        self._next_thread.start()
        return data

    def _callback_epoch_end(self):
        for callback in self._epoch_end_callbacks:

Instances