initialisieren Variablen aus einem restaurierten Modell in TensorFlow

Question

Nehmen wir an, ich die folgende einfache Netzwerk trainieren und speichern (nennen wir es Netzwerk A):

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

Nun würde Ich mag Netzwerk A wiederherzustellen, entwerfen ein neues Netzwerk B und die Parameter des Netzes A die ersten drei Schichten von Netzwerk B initialisieren:

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

Wie kann ich das in TensorFlow?

Answer 1

Sie können einfach Netzwerk A + B erstellen und sich die Conv3-Ebene merken. Sie können dann einen Verlust auf der Ausgabe von Conv3 berechnen, dann für eine Weile trainieren. Sobald Sie zufrieden sind, wechseln Sie einfach zum Training auf einen Verlust auf Conv5.