Капсульная сеть и воспроизводимо - PullRequest
0 голосов
/ 23 октября 2019

Я в настоящее время испытываю трудности при использовании Capsule Network (версия Keras: CapsNet).

Каждый раз, когда я запускаю более 2-5 прогнозов подряд (в цикле), результаты сильно меняются. Я пытался изменить это так сильно. Я также изменил оптимизатор с ADAM на SGD - но я просто не могу сделать его стабильным на 100% и, таким образом, быть в состоянии воспроизвести данный прогон - еще раз.

Как я могу сделать CapsNet воспроизводимым на 100%каждый пробег?

1 Ответ

1 голос
/ 24 октября 2019

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

  1. Установите для переменной среды PYTHONHASHSEED значение0 перед запуском вашей программы на Python.
  2. Если вы выполняете вычисления на графическом процессоре, это может привести к невоспроизводимым результатам из-за округления чисел. Вы можете отключить его и запускать все операции на ЦПУ, задав для переменной среды CUDA_VISIBLE_DEVICES пустую строку таким же образом, как и раньше.
CUDA_VISIBLE_DEVICES="" PYTHONHASHSEED=0 python your_program.py
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...