LoginSignup
5
4

More than 5 years have passed since last update.

TensorFlow v0.12でtf.train.Saverのチェックポイント・フォーマットが変更

Last updated at Posted at 2016-12-14

最新版(v0.12.0rc)にてチェックポイントが読み込めない事による不具合が発生。

v0.12ではtf.train.Saverによって*.metaだけでなく*.indexなどが生成されるなど、生成物の名前にも違いが出る。これにより既存のtf.train.Saverにより生成したチェックポイントを読み込む際に失敗する可能性がある(特に、チェックポイントをファイル名で直接参照しているプログラム)。

対応策

プログラムでは、きちんとtf.train.Saver.last_checkpointsによりチェックポイントのリストを取得する、またはtf.train.latest_checkpointによりlatestに指定されたチェックポイントのパスを取得すること。

なお、暫定の対応としてtf.train.Saverのコンストラクタにwrite_version=1を指定することで既存(V1)のフォーマットを使用できる(v0.12ではデフォルトでwrite_version=2が指定されるように変更されている)。既存コードを移植する際にチェックポイント周りでエラーが起きた場合はwrite_version=1により解決する場合がある。

ちなみに、数日前にTensorFlowのAPIリファレンスのデザインが一新された。

Saverはこちら

References

[ISSUE] seq2seq checkpoints not working #6274 https://github.com/tensorflow/tensorflow/issues/6274

5
4
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
5
4