【发布时间】:2019-12-25 04:28:26
【问题描述】:
Settings:
Mac OS 10.14.6
Python 3.7.4
Tensorflow 2.0.0
我在 name_scope 设置方面遇到了问题。
我在我的代码中写了name_scope(),但它在Tensorboard 中没有像this image in Tensorboard 这样的name_scope。
我打算为这张图片中的 Flatten、Dense 等设置一个名称范围。
我卸载并重新安装了 Tensorflow,但得到了相同的结果。 作为附加信息,我在终端中没有收到任何错误。 有人对这种情况有任何想法吗?
def _model_B0001(self):
with tf.name_scope('1stLayer'):
inputs = Input(self.shape)
x = Flatten()(inputs)
x = Dense(512)(x)
output1 = Dense(self.nb_classes1, activation='softmax', name='output1')(x)
output2 = Dense(self.nb_classes2, activation='softmax', name='output2')(x)
predictions = [output1, output2]
model = Model(inputs, predictions)
model.compile(loss={'output1': 'categorical_crossentropy',
'output2': 'categorical_crossentropy'},
optimizer='adam',
metrics=['accuracy'])
model.summary()
with open(self.summary_txt, "w") as fp:
model.summary(print_fn=lambda x: fp.write(x + "\r\n"))
return model
【问题讨论】:
标签: python tensorflow tensorboard