如何使用多个分支保存/加载keras模型

时间:2017-07-27 10:02:29

标签: merge save keras

我定义了一个keras(2.0.6)模型,它有两个分支合并这些

 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 个答案:

答案 0 :(得分:1)

不要为此使用Sequential API,而是希望Functional API使用分支构建这样的模型。然后你将只有一个模型,它将被保存。