Удаление узлов из графика или сброс всего графика по умолчанию


при работе с глобальным графом по умолчанию можно ли удалить узлы после их добавления или сбросить график по умолчанию на пустой? При работе с TF в интерактивном режиме в IPython мне приходится повторно перезапускать ядро. Я хотел бы иметь возможность экспериментировать с графиками более легко, если это возможно.

3 58

3 ответа:

обновление 11/2/2016

tf.reset_default_graph()

старые вещи

здесь reset_default_graph, но не часть публичного API (я думаю, что это должно быть, кто-то хочет проблема на GitHub?)

моя работа-вокруг, чтобы сбросить вещи это:

from tensorflow.python.framework import ops
ops.reset_default_graph()
sess = tf.InteractiveSession()

по умолчанию сеанс строится вокруг графика по умолчанию. Чтобы избежать оставления мертвых узлов в сеансе, необходимо либо управлять графом по умолчанию, либо использовать явный граф.

  • снимите график по умолчанию, вы можете использовать tf.reset_default_graph

ячейки записной книжки IPython / Jupyter сохраняют состояние между запусками ячейки.

создать пользовательский график:

def main():
    # Define your model
    data = tf.placeholder(...)
    model = ...

with tf.Graph().as_default():
    main()

после запуска график очищается.