将cntk模型v1转换为cntk模型v2

问题描述 投票:0回答:1

我已经使用cntk脑脚本训练了一个模型。然后,我需要在python中使用我的模型。问题在于,使用brainscript保存的模型是v1模型,并且保存了输入节点和输出节点。另一方面,在cntk python api中使用的模型是v2模型。

是否有将v1模型转换为v2模型格式的方法?

如果没有,有一种方法可以从版本1格式中删除输入和输出节点,这样我就可以将特征直接传递给模型的第一层,如下所示:

mode= load_model("cntk_v1.model")
model = mode.clone(method = 'freeze' )
f = cntk.ops.sequence((input_dim))
z = model(f)
deep-learning cntk
1个回答
0
投票

是的,有一种方法可以替换输入输出节点。

下面的示例代码:

new_input = C.input_variable(...)
model = C.load_model(...)
nodes_replacement = {model.arguments[0]: new_input, ANY_NODE_YOU_WANT_REPLACED: NEW_NODE}
new_model = model.clone(C.CloneMethod.clone, nodes_replacement)  # done!
© www.soinside.com 2019 - 2024. All rights reserved.