エクスポート→saved_model (TensorFlow)について

89 views
Skip to first unread message

佐野元昭

unread,
Jan 5, 2024, 10:51:45 PM1/5/24
to Neural Network Console Users (JP)
お世話になります。

Neural Network Consoleを利用させて頂いておりますが、
学習した結果を
TensorFlowのsaved_model形式でエクスポートし、
以下のようなPythonコード:

import tensorflow as tf
from tensorflow import keras
from keras.models import load_model

model = load_model("./saved_model")
model.summary()

で処理すると、

WARNING:tensorflow:SavedModel saved prior to TF 2.5 detected when loading Keras model. Please ensure that you are saving the model with model.save() or tf.keras.models.save_model(), *NOT* tf.saved_model.save(). To confirm, there should be a file named "keras_metadata.pb" in the SavedModel directory.

のようなエラーが出ます。確かに、出力されたsaved_model/には
keras_metadata.pbは存在しません。
ちなみに、当方のTensorFlowは2.15.0ですが、それがまずいのでしょうか。

また、result.nnpを
nnabla_cli convert result.nnp saved_model --export-format SAVED_MODEL
のようにしても、saved_model/は生成されますが、
同様にkeras_metadata.pbは存在しません。

ネットで色々調べましたが、解決しませんでしたので、
アドバイス等をお願いできれば幸いです。

なお、生成されたresult.nnpを用いて
nnabla_cli convert result.nnp result.h5 --nnp-parameter-h5
のようにすると、
 [nnabla][INFO]: Converting: result.h5 successfully!
のように表示されますが、result.h5は見当たりません。
nnabla_cli convert の使い方が誤っているでしょうか。

以上、よろしくお願い致します。

Reply all
Reply to author
Forward
0 new messages