koth/kcws

关于模型导出--output_node_names "transitions,Reshape_9" "transitions,Reshape_7" 什么意思

forever1dream opened this issue · 3 comments

模型导出时指定 output node 在解码的时候作为模型的输出;
训练的时候不是应该指定这两个名字吗?
我在bilstm.py 文件找到了 Reshape_7 这个output的定义
但没找到pos训练 Reshape_9 这个output的定义 以及transitions的定义,
这两个是tensorflow 默认的output node还是什么?
麻烦解释下,谢谢

好像是写死的: kcws/cc/pos_tagger.cc:297: std::vectorstd::string output_names({"Reshape_9"});

koth commented

可以改下训练代码,指定输出节点名,导出时也制定相同名字就好

@koth 意思现在用的是默认参数吗?我好像没有看到指定输出节点的代码。