<< 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()) # これで重みをコピーできる。