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.4/api_docs/python/tf/config/experimental/set_device_policy