在Frozen Graph Tensorflow中替换Concat Op

时间:2018-05-24 00:37:00

标签: python tensorflow lstm frozen tensorrt

民间,

我使用Tensorflow的LSTMCell方法创建了一个冻结图。 我的问题是LSTM Cell使用串联操作。由于我不能解释的原因,我无法使用Concat。 保存冻结图后,有没有办法在冻结图或检查点中替换ConcatV2或连接操作?

1 个答案:

答案 0 :(得分:0)

冻结图用于推理,不需要图的任何中间结果/渐变或推理不需要的参数。 这就是冻结文件(.pb)大小较小的原因,或者如果我们比较元图+数据文件的集合大小,可以说是优化的。

这些经过训练的参数(.data)作为常量存储在冻结图中,并且没有其他元数据或中间参数,这可能需要恢复模型并继续进行训练。

这就是我可以说冻结图无法进一步训练或编辑的原因。