NAN иногда для одного и того же изображения при обучении моей модели тензорного потока yolo для потери

Экземпляр успехаЯ запускаю свою пользовательскую сеть YOLO (вы смотрите только один раз), реализованную исключительно в Тензор-поток. Однако расчет потерь для одного и того же изображения меняется каждый раз, когда я загружаю его в сеть. Иногда уверенность и ширина становятся NAN. Любое предложение о том, как отладить это и что может быть возможной причиной такого поведения? Ошибка экземпляра Я проверил, что входное изображение находится в диапазоне 0–1. Тип данных заполнителей — float 32.


person user8893068    schedule 04.08.2019    source источник


Ответы (1)


Инициализатор ядра сети был ограничен выводом в диапазоне 0-1. Это решило мою проблему с NAN.

person user8893068    schedule 05.08.2019