tf.config.experimental.set_device_policy

Устанавливает текущую политику устройства потока.

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.config.experimental.set_device_policy

tf.config.experimental.set_device_policy(
    device_policy
)

Политика устройства управляет тем, как операции, требующие входных данных на определенном устройстве (например, на GPU:0), обрабатывают входные данные на другом устройстве (например, GPU:1).

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

Эта функция устанавливает политику устройства только для текущего потока. Любой последующий запущенный поток снова будет использовать политику по умолчанию.

Аргументы

Аргументы
`device_policy`	Политика устройства. Допустимые значения: None: Переключение на системный по умолчанию. 'warn': Копирует тензоры, которые не находятся на правильном устройстве, и выводит предупреждение. 'explicit': Вызывает ошибку, если размещение не соответствует требованиям. 'silent': Беззвучно копирует тензоры. Обратите внимание, что это может скрыть проблемы производительности, так как уведомления о блокировании операций на тензоре, копируемом между устройствами, отсутствуют. 'silent_for_int32': беззвучно копирует `int32` тензоры, вызывая ошибки для других.

device_policy

Политика устройства. Допустимые значения:

None: Переключение на системный по умолчанию.
'warn': Копирует тензоры, которые не находятся на правильном устройстве, и выводит предупреждение.
'explicit': Вызывает ошибку, если размещение не соответствует требованиям.
'silent': Беззвучно копирует тензоры. Обратите внимание, что это может скрыть проблемы производительности, так как уведомления о блокировании операций на тензоре, копируемом между устройствами, отсутствуют.
'silent_for_int32': беззвучно копирует int32 тензоры, вызывая ошибки для других.

Возбуждает исключения
`ValueError`	Если передано недопустимое значение `device_policy`.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/config/experimental/set_device_policy