Tensorflowでパーティション変数を保存および復元する方法

Question

大きな行列があります。

私はこの変数をシャードの数として作成します。

softmax_w = tf.get_variable("softmax_w", [hps.vocab_size, hps.projected_size], 
          partitioner=tf.fixed_size_partitioner(hps.num_shards, 0)) 

ログを作成します。

model/softmax_w/part_0:0 (99184, 512) /cpu:0 
model/softmax_w/part_1:0 (99184, 512) /cpu:0 
model/softmax_w/part_2:0 (99184, 512) /cpu:0 
model/softmax_w/part_3:0 (99184, 512) /cpu:0 
model/softmax_w/part_4:0 (99184, 512) /cpu:0 
model/softmax_w/part_5:0 (99184, 512) /cpu:0 
model/softmax_w/part_6:0 (99183, 512) /cpu:0 
model/softmax_w/part_7:0 (99183, 512) /cpu:0 

私はそれを成功さトレーニングや保存することができます。しかし、私がモデルを復元しようとすると、私はこのエラーを受け取りました：

W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_7 not found in checkpoint 
W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_6 not found in checkpoint 
W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_5 not found in checkpoint 
W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_4 not found in checkpoint 
W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_3 not found in checkpoint 
W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_2 not found in checkpoint 
W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_1 not found in checkpoint 
W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_0 not found in checkpoint 
W tensorflow/core/framework/op_kernel.cc:975] Not found: Key model/softmax_w/part_7 not found in checkpoint 

私はtensorflow変数を一部として保存しました。保存されたパラメータの値はsoftmax_wです。もはやパーティション化された変数ではありません。

Answer 1

テンソルフロー0.12で発生し、1.3（2017年10月の最終バージョン）で発生しません。ここには同じ著者によって提出され、今修正されたGitHub issueがあります。このエラーが表示された場合は、テンソルフローをアップグレードしてください。