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

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


表示用於自動調整數據集性能的選項。

用法

tf.data.experimental.AutotuneOptions()

屬性

  • autotune_algorithm 啟用自動調整時(通過 autotune )，確定要使用的算法。
  • cpu_budget 啟用自動調整時(通過 autotune )，確定要使用的 CPU 預算。允許大於可調度 CPU 內核數的值，但可能會導致 CPU 爭用。如果為 None，則默認為可調度的 CPU 內核數。
  • enabled 是否自動調整性能旋鈕。如果沒有，默認為真。
  • ram_budget 啟用自動調整時(通過 autotune )，確定要使用的 RAM 預算。大於可用 RAM(以字節為單位)的值可能會導致 OOM。如果為 None，則默認為可用 RAM 的一半(以字節為單位)。 
options = tf.data.Options()
options.autotune.enabled = False
dataset = dataset.with_options(options)

相關用法


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