0
を指定せずにtensorflowで変数を復元しますか?これは、変数をロードして分類を実行するために変数を使用する場合に非常に役立ちます。は、私はtensorflowに変数をこのように復元することができ、その形状
を指定せずにtensorflowで変数を復元しますか?これは、変数をロードして分類を実行するために変数を使用する場合に非常に役立ちます。は、私はtensorflowに変数をこのように復元することができ、その形状
あなたがチェックポイントからメタグラフをロードすることでこれを行うことができます。
with tf.Session() as sess:
saver = tf.train.import_meta_graph(
'my-save-dir/my-model-10000.meta')
saver.restore(sess, 'my-save-dir/my-model-10000')
をこれがモデルを保存するときwrite_meta_graph=True
を使用する必要があります - それはしかし、デフォルトです。
この詳細については、hereをご覧ください。