Реализация mask-r-cnn с распределенным тензорным потоком
Я тренирую сеть mask-r-cnn, которая построена на тензорном потоке и керасе. Я ищу способ сократить время обучения, поэтому я подумал реализовать его с помощью tenorflow-distribution.
Я работаю с mask-r-cnn в течение некоторого времени, но, похоже, то, что я пытаюсь сделать, потребует от меня изменения исходного кода mask-r-cnn, который выше моих нынешних навыков.
Итак, мой вопрос: кто-нибудь когда-либо делал это или что-то подобное? возможно ли это вообще, или я неправильно понимаю использование тензорного потока.
Спасибо вперед.
1 ответ
Даже без распределенного обучения реализация тензорного пакета Mask R-CNN в тензорном потоке работает в 5 раз быстрее (и также более точно), чем та, к которой вы привязаны.
Он также поддерживает распределенное обучение с MPI.