我正在编写一个优化损失函数的模型,该函数由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])