Skip to content

S
seminar-breakout
Commit da062119 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix bug in PrefetchDataZMQ

parent 68e8d9eb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tensorpack/dataflow/prefetch.py
View file @ da062119
...@@ -247,8 +247,9 @@ class PrefetchDataZMQ(_MultiProcessZMQDataFlow): ...@@ -247,8 +247,9 @@ class PrefetchDataZMQ(_MultiProcessZMQDataFlow):
nr_proc (int): number of processes to use. nr_proc (int): number of processes to use.
hwm (int): the zmq "high-water mark" (queue size) for both sender and receiver. hwm (int): the zmq "high-water mark" (queue size) for both sender and receiver.
""" """
super(PrefetchDataZMQ, self).__init__(ds) super(PrefetchDataZMQ, self).__init__()
self.ds = ds
self.nr_proc = nr_proc self.nr_proc = nr_proc
self._hwm = hwm self._hwm = hwm
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment