Участок Keras Custom Loss в Тензорборде - PullRequest
Новая
       47

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

2 голосов
/ 16 января 2020

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

Здесь я добавляю пример моего пользовательского кода функции потери. То, что я хочу здесь, это отобразить это 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()

1 Ответ

0 голосов
/ 16 января 2020

Добрый день! Возможно, такая реализация может решить проблему: 

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
...