相关疑难解决方法(0)

如何在Tensorflow-2.0中绘制tf.keras模型图?

我已升级到Tensorflow 2.0,没有tf.summary.FileWriter("tf_graphs", sess.graph)。我正在研究有关此问题的其他StackOverflow问题,他们说可以使用tf.compat.v1.summary etc。当然,必须有一种在Tensorflow版本2中绘制和可视化tf.keras模型的方法。这是什么?我正在寻找以下张量板输出。谢谢!

在此处输入图片说明

python-3.x tensorflow tensorboard tf.keras tensorflow2.0

7
推荐指数
2
解决办法
925
查看次数

TensorFlow 2.0中图可视化的简单案例

我想简单地定义一个模型,并在TensorBoard中可视化其图形以进行初始体系结构检查。因此,我不想为此目的进行任何计算。

在TensorFlow 1.X中,在一个tf.Session()我可以简单地flush()编写摘要文件的地方轻松实现。

在TensorFlow 2.0中没有tf.Session(),因此问题是如何实现呢?

以下是示例代码。为了在TensorBoard中编写图结构,我需要添加什么?

from nets import i3d
import tensorflow as tf

def i3d_output(model, x):
    out, _ = model(x)
    return out

tf.compat.v1.disable_eager_execution()
x = tf.random.uniform(shape=(4,179,224,224,3))
model = i3d.InceptionI3d()
net = i3d_output(model, x)
train_summary_writer = tf.summary.create_file_writer('/home/uujjwal/bmvc2019')
Run Code Online (Sandbox Code Playgroud)

tensorflow tensorflow2.0

6
推荐指数
1
解决办法
531
查看次数