Участок Keras Custom Loss в Тензорборде

Я использую кераты с бэкэндом tenorflow 1.9, и я сделал свою собственную функцию потерь. Я запутался, как построить свою собственную функцию потерь в тензорной доске. Я попробовал несколько примеров, но ни один из них не нанесен на тензорную доску.

Здесь я добавляю пример моего пользовательского кода функции потери. То, что я хочу здесь, это построить эти убыток_1, убыток_2 и убыток_в итоги в тензорной доске. Кто-нибудь знает как это делать?

Спасибо

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()

Всего 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

Есть идеи?

10000