在 Tensorboard 中绘制 Keras 自定义损失答案

【问题标题】：Plot Keras Custom Loss in Tensorboard在 Tensorboard 中绘制 Keras 自定义损失
【发布时间】：2020-05-03 09:36:18
【问题描述】：

我正在使用带有 tensorflow 1.9 后端的 keras，并且我已经制作了我的自定义损失函数。我很困惑如何将我的自定义损失函数绘制到张量板上。我已经尝试了几个示例，但没有一个可以绘制到 tensorboard 中。

这里我添加了我的自定义损失函数代码的例子。我在这里想要的是在张量板上绘制这个 loss_1、loss_2 和 loss_total。有人知道怎么做吗？

谢谢

def loss_1():
   return K.mean(-0.5*var_1)

def loss_2(x, var2):
   return K.mean(0.5*x + var_2)

def loss_total(x, var_z):
   return 0.5 * loss_2(x, var_z) + loss_1()

【问题讨论】：

欢迎来到 SO。这是一个很好的问题。但是，如果您可以添加迄今为止尝试过的内容，这可能是一个更好的问题。
好吧，首先我尝试将自定义损失放入 keras .fit 但它给了我错误，其次我尝试创建一个定义我的自定义损失并放入 .fit 的类，但仍然 tensorboard 赢了不要绘制它
您能否将代码作为问题的一部分发布。
查看我的更新

标签： tensorflow keras deep-learning tensorboard loss-function

【解决方案1】：

下午好！或许这样的实现可以解决问题：

from keras.callbacks import TensorBoard
#create tensorboard callback
tensorboard = TensorBoard(log_dir=os.path.expanduser('./YOUR_LOG_FOLDER/some_name'), 
                          write_graph=True
                          #another code
                          )

YOUR_MODEL.compile(loss=custom_loss, optimizer=YOUR_OPTIMIZER)

#Specify tensorboard callback
YOUR_MODEL.fit(
                #another code
                callbacks = [tensorboard], 
                #another code
                )

在另一个终端运行

tensorboard --logdir=YOUR_LOG_FOLDER

【讨论】：

嗨，我已经这样做了，但它仍然不会绘制我的自定义损失
你打开了地址localhost:6006的tensorboard吗？