输出节点张量流检查点的名称

时间:2019-04-10 14:05:01

标签: tensorflow

我有一些tensorflow检查点文件。现在,我试图冻结模型并创建一个.pb文件。为此,我需要输出节点名称。我得到了节点名称列表,但是我无法弄清楚输出节点名称。从网络体系结构中,我发现模型的输出具有完全连接的层。我的变量列表具有以下名称(这是一个很长的列表,因此这里只保留与FC层相关的那些名称),

('fully_connected/biases', [2]),  
('fully_connected/biases/Adam', [2]),  
('fully_connected/biases/Adam_1', [2]),  
('fully_connected/weights', [256, 2]),  
('fully_connected/weights/Adam', [256, 2]),  
('fully_connected/weights/Adam_1', [256, 2])  

由此,我应该在函数“ tf.freeze_graph”中指定的确切名称是什么?

0 个答案:

没有答案