這個 API 還在實驗階段，只有 tf-nightly 版本才支援。

VariablePolicy

政策

exported = tf.train.Checkpoint()

with tf.device('/GPU:0'):
  exported.x_gpu = tf.Variable(1.0)

with tf.device('/CPU:0'):
  exported.x_cpu = tf.Variable(1.0)

tf.saved_model.save(exported, export_dir,
    options = tf.saved_model.SaveOptions(
        experimental_variable_policy=
          tf.saved_model.experimental.VariablePolicy.SAVE_VARIABLE_DEVICES))

類別變數

• EXPAND_DISTRIBUTED_VARIABLES
• NONE
• SAVE_VARIABLE_DEVICES