dd2d9ffaa04e6bfd7cf3c1dbd605ffe50b61f7bf,tensorpack/dataflow/parallel.py,MultiThreadPrefetchData,__init__,#MultiThreadPrefetchData#Any#Any#Any#,398

Before Change

        
        assert nr_thread > 0 and nr_prefetch > 0
        self.nr_thread = nr_thread
        self.queue = queue.Queue(maxsize=nr_prefetch)
        self.threads = [
            MultiThreadPrefetchData._Worker(get_df, self.queue)
            for _ in range(nr_thread)]

    def reset_state(self):
        for th in self.threads:
            th.df.reset_state()

After Change

    class _Worker(mp.Process):
        def __init__(self, ds, queue, idx):
            super(MultiProcessRunner._Worker, self).__init__()
            self.ds = ds
            self.queue = queue
            self.idx = idx

Instances