tf.data.experimental.service.CrossTrainerCache
Параметры, относящиеся к кэшу tf.data service для совместного использования между тренерами.
tf.data.experimental.service.CrossTrainerCache(
trainer_id
)
Используется для включения кэша совместного использования между тренерами при распределении набора данных. Например:
dataset = dataset.apply(tf.data.experimental.service.distribute(
processing_mode=tf.data.experimental.service.ShardingPolicy.OFF,
service=FLAGS.tf_data_service_address,
job_name="job",
cross_trainer_cache=data_service_ops.CrossTrainerCache(
trainer_id=trainer_id())))
Подробнее см. в разделе https://www.tensorflow.org/api_docs/python/tf/data/experimental/service#sharing_tfdata_service_with_concurrent_trainers
| Аргументы | |
|---|---|
trainer_id | Каждое задание обучения имеет уникальный идентификатор. После того как задание использовало данные, они остаются в кэше и повторно используются заданиями с другими trainer_idами. Запросы с одинаковым trainer_id не используют данные повторно. |
| Вызывает исключения | |
|---|---|
ValueError, если trainer_id пуст. |
© 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/api_docs/python/tf/data/experimental/service/CrossTrainerCache