我已经使用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)
答案 0 :(得分: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!