テンソルフローモデルチェックポイントからのウェイト値の抽出

Question

私はテンソルフローでモデルをトレーニングしています。私は自分のモデルのチェックポイントを実行しています。 I私は今、私は私のグラフの各レイヤーの重み値を抽出するために、4つのファイル、すなわち、

• checkpoint
• model.cpkt-0.data-00000-of-00001
• model.cpkt-0.index
• model.cpkt-0.meta

をしたいしているCheckpointsディレクトリ、 、 どうやってやるの？

私はこの試みた：

import tensorflow as tf 
sess = tf.InteractiveSession() 

saver = tf.train.import_meta_graph('model.cpkt-0.meta') 
w = saver.restore(sess, 'model.cpkt-0.data-00000-of-00001') 

をしかし、私は次のエラーを取得しています：

Unable to open table file ./model.cpkt-0.data-00000-of-00001: Data loss: not an sstable (bad magic number): perhaps your file is in a different file format and you need to use a different restore operator? 

Answer 1

あなたは間違った方法で

saver.restore(sess, 'model.cpkt-0') 
# get the graph 
g = tf.get_default_graph() 
w1 = g.get_tensor_by_name('some_variable_name as per your definition in the model') 

を復元します