Как включить ведение журнала размещения устройств (log_device_placement) в TensorFlow 2.n - PullRequest
Новая
       84

Как включить ведение журнала размещения устройств (log_device_placement) в TensorFlow 2.n

1 голос
/ 29 апреля 2020

Я хотел бы посмотреть, какие операции выполняются на каком устройстве. В TensorFlow 1.n можно использовать конфигурацию сеанса (log_device_placement). Есть ли аналогичная операция в TF 2.n?

1 Ответ

1 голос
/ 03 мая 2020

Чтобы определить, какие операции выполняются на каком устройстве, используйте 

tf.debugging.set_log_device_placement(True)

в качестве первой строки после импорта TensorFlow. Подробнее вы можете найти здесь

...