Инициализация переменных из восстановленной модели в TensorFlow

Question

Предположим, что я тренируюсь и сохранить следующую простую сеть (Назовём это сеть A):

Input --> Conv1 --> Conv2 --> Conv3 --> Output 

Теперь я хотел бы восстановить сеть А, создать новую сеть B и использовать параметры сети а инициализировать первые 3 слоя сети B:

Input --> Conv1 --> Conv2 --> Conv3 --> Conv4 --> Conv5 --> Output 

Как я могу сделать это в TensorFlow?

Answer 1

Вы можете просто создать сеть A + B и запомнить уровень Conv3. Затем вы можете вычислить потерю на выходе Conv3, а затем тренироваться некоторое время. Как только вы удовлетворитесь, просто переключитесь на тренировку с потерей на Conv5.