tf.keras.utils.OrderedEnqueuer
View source on GitHub |
Builds a Enqueuer from a Sequence.
Inherits From: SequenceEnqueuer
tf.keras.utils.OrderedEnqueuer( sequence, use_multiprocessing=False, shuffle=False )
Used in fit_generator
, evaluate_generator
, predict_generator
.
Arguments | |
---|---|
sequence | A tf.keras.utils.data_utils.Sequence object. |
use_multiprocessing | use multiprocessing if True, otherwise threading |
shuffle | whether to shuffle the data at the beginning of each epoch |
Methods
get
get()
Creates a generator to extract data from the queue.
Skip the data if it is None
.
Yields | |
---|---|
The next element in the queue, i.e. a tuple (inputs, targets) or (inputs, targets, sample_weights) . |
is_running
is_running()
start
start( workers=1, max_queue_size=10 )
Starts the handler's workers.
Arguments | |
---|---|
workers | Number of workers. |
max_queue_size | queue size (when full, workers could block on put() ) |
stop
stop( timeout=None )
Stops running threads and wait for them to exit, if necessary.
Should be called by the same thread which called start()
.
Arguments | |
---|---|
timeout | maximum time to wait on thread.join() |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/utils/OrderedEnqueuer