Частичное восстановление модели в Tensorflow

#tensorflow #deep-learning

#tensorflow #глубокое обучение

Вопрос:

У меня есть 2 модели, архитектуры которых отличаются только последним уровнем FC. Мой вопрос: как я могу инициализировать веса второй модели с помощью весов, полученных для первой модели (без явной передачи имен каждой переменной (кроме переменных последнего уровня FC) в saver.restore)? Спасибо.

Ответ №1:

Это руководство по tensorflow отвечает на вашу проблему:https://www.tensorflow.org/versions/master/how_tos/image_retraining/index.html