Initialiser les variables d'un modèle restauré dans tensorflow

Question

Supposons que je train et enregistrer le réseau simple suivant (permet de l'appeler réseau A):

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

Maintenant, je voudrais rétablir le réseau A, la conception d'un nouveau réseau B et paramètres utilisation du réseau A pour initialiser les 3 premières couches de réseau B:

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

Comment puis-je faire cela en tensorflow?

Answer 1

Vous pouvez simplement créer le réseau A + B et vous rappeler la couche Conv3. Vous pouvez alors calculer une perte sur la sortie de Conv3, puis vous entraîner pendant un moment. Une fois que vous êtes satisfait, il suffit de passer à l'entraînement sur une perte sur Conv5.