Keras的其他标量摘要

时间:2018-01-04 23:00:20

标签: keras tensorboard

我正在编写一个优化损失函数的模型,该函数由5部分组成

def custom_loss(y_true, y_pred):
    ...

    loss = 5. * loss_xy + 5. * loss_wh + loss_conf + 0.5 * loss_notconf + loss_class
    return loss

稍后在代码中,当我运行fit_generator时,我传递Tensorboard回调,它允许我查看我的损失函数的行为。我无法弄清楚的是如何在Tensorboard中看到最终损失函数的各个部分(loss_xy,loss_w,loss_conf,loss_notconf,loss_class)。有没有正确的方法呢?我在文档中找不到任何内容

tensorboard = callbacks.TensorBoard(log_dir=log_dir, write_graph=True)
tensorboard.set_model(model)

model.fit_generator(generator=train_iterator, epochs=50, callbacks=[tensorboard])

0 个答案:

没有答案