Keras-CNTK 保存模型-v2 格式

Question

我使用 CNTK 作为 Keras 的后端。我正在尝试使用我在 C++ 中使用 Keras 训练过的模型。

我已经使用 HDF5 中的 Keras 训练并保存了我的模型。我现在如何使用 CNTK API 将其保存为 model-v2 格式？

我试过这个:

model = load_model('model2.h5')
cntk.ops.functions.Function.save(model, 'CNTK_model2.pb')

但我收到以下错误:

TypeError: save() missing 1 required positional argument: 'filename'

如果 tensorflow 是后端，我会这样做:

model = load_model('model2.h5')
sess = K.get_session()
tf_saver = tf.train.Saver()
tf_saver.save(sess=sess, save_path=checkpoint_path)

我怎样才能达到同样的目的？

Answer 1

根据评论here ，我能够使用这个:

import cntk as C
import keras.backend as K

keras_model = K.load_model('my_keras_model.h5')

C.combine(keras_model.model.outputs).save('my_cntk_model')
cntk_model = C.load_model('my_cntk_model')