Можно ли напечатать значение облачной переменной TPU в тензорном потоке?

Question

Отключив активное выполнение, я могу подключиться к своему облачному TPU и запустить свой цикл обучения. После расчета потерь я бы хотел напечатать эту переменную.

Учитывая, что потеря - это тензор облачного TPU, я до сих пор не нашел способа распечатать его. tf.print возвращает PrintOperation, поэтому я застрял. Я предполагаю, что что-то вроде перемещения тензора потерь обратно в мой процессор будет работать, но я не нашел никакого хакерского способа сделать это.

Я знаю, что можно получить этот результат, потому что когда я инкапсулирую свою модель сkeras в соответствии со стратегией распространения TPU и использованием model.fit вместо моего пользовательского цикла обучения, я могу напечатать показатели потерь.

Так что должен быть способ, и любая помощь в поиске этого очень ценится:)

Answer 1

Хорошо, найди мой хакерский способ, даже если бы существовали лучшие альтернативы, я бы хотел их услышать:

sess = tf.compat.v1.Session()
init_op = tf.compat.v1.global_variables_initializer()
sess.run(init_op)
<tensor>.eval(session=sess)