MCPcopy Index your code
hub / github.com/tensorpack/tensorpack / EnqueProc

Class EnqueProc

tensorpack/dataflow/remote.py:182–196  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

180 q = mp.Queue(size)
181
182 class EnqueProc(mp.Process):
183
184 def __init__(self, df, q, nr_consumer):
185 super(EnqueProc, self).__init__()
186 self.df = df
187 self.q = q
188
189 def run(self):
190 self.df.reset_state()
191 try:
192 for idx, dp in enumerate(self.df):
193 self.q.put((idx, dp))
194 finally:
195 for _ in range(nr_consumer):
196 self.q.put((DIE, None))
197
198 proc = EnqueProc(df, q, nr_consumer)
199 return q, proc

Callers 1

Calls

no outgoing calls

Tested by

no test coverage detected