tenssorflowはckptの各ノードの名称の実例を調べます。
537 ワード
下記のスクリプトを実行すると、モデルのノード変数名が印刷されます。
以上のこのtenssorflowはckptの各ノードの名称の実例を調べて、小さい編集がみんなのすべての内容に分かち合うので、みんなに1つの参考をあげることができることを望んで、みんながよけいに私達を支持することをも望みます。
from tensorflow.python import pywrap_tensorflow
import os
checkpoint_path=os.path.join('model.ckpt-131805')
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print 'tensor_name: ',key
以上のこのtenssorflowはckptの各ノードの名称の実例を調べて、小さい編集がみんなのすべての内容に分かち合うので、みんなに1つの参考をあげることができることを望んで、みんながよけいに私達を支持することをも望みます。