2017-07-27 1 views
1

je définir un keras (2.0.6) modèle avec deux branches de la fusion de cesComment enregistrer/charger le modèle du keras avec plusieurs branches

left=Sequential() 

    ... 

    right=Sequential() 

    ... 

    model = Sequential() 

    model.add(Merge([left,right], mode='concat')) 

model.save('mymodel.model') 

Puis, plus tard si je veux réutiliser mon modèle après model = load('mymodel.model') je reçois Il semble donc que la commande save n'a enregistré que la partie 'model' de mon modèle, pas les branches gauche et droite. Comment peut-on sauver le modèle entier? Ou aurai-je besoin de trois fichiers modèles, un pour la branche gauche, un pour le modèle droit et un pour le modèle fusionné?

Il semble qu'il y ait eu une discussion à ce sujet précédemment (https://github.com/fchollet/keras/issues/1349) mais il n'y avait pas de solution, pour autant que je puisse voir.

Est-ce que quelqu'un sait si/comment cela a été résolu?

Répondre

1

N'utilisez pas l'API séquentielle pour cela, préférez le Functional API pour construire un tel modèle avec des branches. Ensuite, vous n'aurez qu'un seul modèle et il sera sauvegardé en tant que tel.