覚え書きブログ

読者です 読者をやめる 読者になる 読者になる

Tensorflow v1.0.0への移行

ディープラーニング

Tensorflow v1で、v0.12で書いたコードを実行すると下記のようなモジュールにpackが無いとか、Splitのsplit_dimの入力がおかしいなどのエラーがでる。

AttributeError: 'module' object has no attribute 'pack'
TypeError: Input 'split_dim' of 'Split' Op has type float32 that does not match expected type of int32.

これは、v1にて「pack」が「split」に関数名が変更されたり、splitの入力変数の順番が変更になったのが原因である。以下のサイトにて、自動的にv1にアップグレードしてくれるコードや、手動で関数名を変更する方法が紹介されている。
https://www.tensorflow.org/install/migration
Transitioning to TensorFlow 1.0  |  TensorFlow