tf.saved_model.experimental.VariablePolicy

Перечисление, определяющее варианты обработки переменных при сохранении.

NONE Политика не применена: распределённые переменные сохраняются как одна переменная без привязки к устройству.

SAVE_VARIABLE_DEVICES При сохранении переменных также сохраняется назначение их устройств. Это полезно, если требуется жёстко задать устройства в сохранённых моделях, но это также делает их непереносимыми, если отключение мягкого размещения устройств отключено (подробнее в tf.config.set_soft_device_placement). В настоящее время это не полностью поддерживается saved_model.load и в основном предназначено для использования при чтении сохранённой модели на более низком уровне API. В примере ниже граф, сохранённый вызовом saved_model.save, будет содержать правильно указанные устройства переменных:

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 Распределённые переменные будут сохраняться с информацией о своих компонентах, что позволит восстановить их при загрузке. Кроме того, сохранённый граф будет содержать ссылки на эти переменные. Это полезно, когда требуется использовать модель для обучения в средах, где исходная стратегия распределения недоступна.