Две разные функции потерь с разными метками в тензорном потоке - PullRequest
0 голосов
/ 25 октября 2018

Я тренирую сеть с двумя путями, каждый из которых состоит из нескольких слоев.Один путь относится к видеовходу, а другой путь - к аудиовходу.В какой-то момент я смешиваю функции, а затем применяю некоторые другие конвективные слои и полностью связанные слои к выходу сети.

Я хочу определить две отдельные функции потерь: loss1 непосредственно перед микшированием звука иособенности видео и loss2 на выходе.Проблема в том, что метки разные для каждой функции оптимизации.У меня есть две метки класса для слоя микса, которые указывают, совпадают ли видео и аудио, и еще две метки класса в конце сети, которые указывают на определенные действия в видео.

Любые предложения о том, как я долженопределить две мои функции потери, оптимизаторы и обучить сеть в тензор потока?Обратите внимание, что хотя метки и функции потери различны, большая часть сети одинакова.

Спасибо,

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...