今日学んだこと＃0011 - rokkiの備忘録

tensorflow上でモデルの重みの保存と読み込みに，思いの外手間取った．

結論

tf.train.Saver()に重みのリストを渡すと，保存または読み込む．何も渡さないと，すべて保存しようとする．よって，事前に学習した重みを読み込むときは，variableのリストを渡してあげる．

その他Tips

保存したcheckpointの中身を見るためのスクリプト *1

def check_ckpt(save_model_path="./model.ckpt", all_tensors=False):
    from tensorflow.python.tools import inspect_checkpoint as chkp
    chkp.print_tensors_in_checkpoint_file(save_model_path, tensor_name='', all_tensors=all_tensors)

Tensorflow savor.restore()の実装に関する解説記事*2

*1:github.com

*2:blog.csdn.net