tf.keras.utils.GeneratorEnqueuer
| Просмотреть исходный код на GitHub |
Создаёт очередь из генератора данных.
Наследуется от: SequenceEnqueuer
tf.keras.utils.GeneratorEnqueuer(
generator, use_multiprocessing=False, random_seed=None
)
Предоставленный генератор может быть конечным, в этом случае класс выбросит исключение StopIteration.
| Аргументы | |
|---|---|
generator | функция-генератор, которая возвращает данные |
use_multiprocessing | использовать многопроцессорность, если True, в противном случае — потоки |
random_seed | Начальное значение для рабочих потоков. Будет инкрементировано на единицу для каждого потока. |
Методы
get
get()
Создаёт генератор для извлечения данных из очереди.
Пропускает данные, если они None.
| Возвращает | |
|---|---|
Следующий элемент в очереди, т.е. кортеж (inputs, targets) или (inputs, targets, sample_weights). |
is_running
is_running()
start
start(
workers=1, max_queue_size=10
)
Запускает потоки обработки.
| Аргументы | |
|---|---|
workers | Количество потоков. |
max_queue_size | Размер очереди (если она полна, потоки могут заблокироваться на put()) |
stop
stop(
timeout=None
)
Останавливает работающие потоки и ждёт их завершения, если необходимо.
Должен вызываться тем же потоком, который вызвал start().
| Аргументы | |
|---|---|
timeout | максимальное время ожидания на thread.join() |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/utils/GeneratorEnqueuer
