Файл tensorflow .graph не создается

Это пример кода.

import tensorflow as tf

const1 = tf.constant(2)
const2 = tf.constant(3)
add_op = tf.add(const1,const2)
mul_op = tf.mul(add_op,const2)

with tf.Session() as sess:

result,result2 = sess.run([mul_op,add_op])
print(result)
print(result2)

tf.train.SummaryWriter('./',sess.graph)

это показывает сообщение, как это,

Tensor("Add:0", shape=(), dtype=int32)

Однако файл не создается.


person whitebear    schedule 03.06.2017    source источник


Ответы (1)


Вот модифицированный код, который должен работать:

import tensorflow as tf

const1 = tf.constant(2)
const2 = tf.constant(3)
add_op = tf.add(const1,const2)
mul_op = tf.multiply(add_op,const2)  # probably you use old version of TF

with tf.Session() as sess:
    writer = tf.summary.FileWriter('logs', sess.graph)
    result,result2 = sess.run([mul_op,add_op])
    writer.close()

Я удалил отпечатки, изменил название операции, чтобы иметь возможность запускать ее в новой версии TF (рекомендую обновить ) положил писатель сверху и правильно закрыл его в конце. Также измените каталог журнала.

Теперь из того же каталога, который вы использовали для запуска скрипта, выполните следующую команду: tensorboard --logdir=logs. Перейдите в браузер и посмотрите результаты.

person Salvador Dali    schedule 03.06.2017
comment
Спасибо, моя версия была 0.10rc, поэтому я обновлю версию 1. - person whitebear; 03.06.2017