f59fb87d9613da132096e85b60cf6c0a8efc1cf6,python/src/nnabla/utils/data_source_implements.py,CacheDataSource,__init__,#CacheDataSource#Any#Any#Any#Any#,281

Before Change

        if os.path.exists(index_filename):
            self.initialize_cache_files_with_index(index_filename)
        else:
            self._filenames = [f for f in self._filereader.listdir() if os.path.splitext(f)[
                1].lower() == ".h5"]
            for filename in self._filenames:
                self.initialize_cache_files(filename)

        logger.info("{}".format(len(self._cache_files)))

        self._cache_reader_with_prefetch = CacheReaderWithPrefetch(
            self._cachedir, self._num_of_threads, self._variables)

After Change

            self._cachedir, self._num_of_threads, self._variables)
        self._thread_lock = threading.Lock()

        self._original_order = []
        for i in range(len(self._cache_files)):
            filename, length = self._cache_files[i]
            for j in range(length):
                self._original_order.append((filename, j))

        self.reset()

    def reset(self):
        with self._thread_lock:

Instances