當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.data.experimental.DistributeOptions用法及代碼示例


表示分布式數據處理的選項。

用法

tf.data.experimental.DistributeOptions()

屬性

  • auto_shard_policy 要使用的分片類型。有關其他信息，請參閱tf.data.experimental.AutoShardPolicy。
  • num_devices 連接到此輸入管道的設備數量。這將由 MultiDeviceIterator 自動設置。

您可以通過 tf.data.Options 的 experimental_distribute 屬性設置數據集的分布選項；該屬性是 tf.data.experimental.DistributeOptions 的一個實例。

options = tf.data.Options()
options.experimental_distribute.auto_shard_policy = AutoShardPolicy.OFF
dataset = dataset.with_options(options)

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.data.experimental.DistributeOptions。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。