CNTK:访问经过培训的模型数据

时间:2017-01-10 03:30:39

标签: python cntk

我使用Python API在CNTK中训练了一个模型。我想在Android设备上的代码中实现网络。

有没有办法可以访问网络权重,然后直接编码网络而不使用CNTK库?

我可以以人类可读的形式访问模型吗?

2 个答案:

答案 0 :(得分:4)

是的,你确定可以。这些信息可以在CNTK github上找到(“我如何”部分)。

https://github.com/Microsoft/CNTK/wiki/Load-model-and-access-network-weights-(parameters)

答案 1 :(得分:0)

您可以使用CNTK的dumpnode命令将CNTK训练的模型转换为txt格式。这是内容配置文件txt.conf:

command = convert2txt

convert2txt = [
    action = "dumpnode"    
    modelPath="./cntkSpeechFF.dnn.5"
    nodeName = "Prior" # if not specified, all nodes will be printed
    outputFile = "./cntkSpeechFF.dnn.5.txt" # the path to the output file. If not specified a file name will be automatically generated based on the modelPath.

    printValues = true
    printMetadata = true

]

然后您以cntk身份运行

cntk configFile=txt.conf