如何解决如何使用Tensorboard通过Keras功能API创建活动直方图
我使用Keras功能API创建以下网络:
input = Input(shape=input_shape)
x = Conv2D(filters=32,kernel_size=(3,3),activation='relu')(input)
tf.summary.histogram(name="conv1",data=x)
x = Conv2D(filters=64,activation='relu')(x)
tf.summary.histogram(name="conv2",data=x)
x = MaxPool2D(pool_size=(2,2))(x)
x = Flatten()(x)
x = Dense(units=128,activation='relu')(x)
tf.summary.histogram(name="dense1",data=x)
x = Dense(units=num_classes,activation='softmax')(x)
tf.summary.histogram(name="demse1",data=x)
model = Model(inputs=inp,outputs=x)
我用tf.summary.histogram
提取了有关不同层激活的信息。但是,该层的激活直方图不会出现在Tensorboard中。
我该怎么办?
解决方法
您可以使用tensorboard回调获取直方图。
tensorboard_callback = tf.keras.callbacks.TensorBoard(
log_dir='logs',histogram_freq=1,profile_batch = 0
)
model.fit(x,y,epochs = 5,callbacks = tensorboard_callback)
有关完整的文档,请参见here
Tensorboard直方图-
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。