<< 2024-06-05 | 2024-06-07 >>
DailyNote
- TensorflowのSavedModel形式で保存すると単一のファイルではなくフォルダ内に複数のファイルが配置される。
- このフォルダ構造が適切になっていないと、
tf.keras.models.load_model
で呼び出せなくなる。
- このフォルダ構造が適切になっていないと、
- モデルの重みをコピーする場合
import tensorflow as tf
model = tf.keras.models.load_model(model_path)
another_model = tf.keras.models.load_model(another_model_path)
model.set_weights(another_model.get_weights()) # これで重みをコピーできる。