2017-07-27 82 views
1

我定義了一個keras(2.0.6)有兩個分支模型合併這些如何保存/載入keras模型

left=Sequential() 

    ... 

    right=Sequential() 

    ... 

    model = Sequential() 

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

model.save('mymodel.model') 

再後來,如果我想model = load('mymodel.model')我收到之後再利用我的模型

"ValueError: You are trying to load a weight file containing 5 layers into a model with 1 layers.". 

所以看起來save命令只保存了模型的'模型'部分,而不是左右分支。如何保存整個模型?或者我需要三個模型文件,一個用於左側分支,一個用於右側,另一個用於合併模型?

似乎之前一直在討論這個問題(https://github.com/fchollet/keras/issues/1349),但根據我的看法,沒有解決方案。

有誰知道是否/如何解決了這個問題?

回答

1

不要爲此使用Sequential API,更喜歡Functional API來構建具有分支的模型。那麼你將只有一個模型,它將被保存。