Kerasを使用してTensorflow 2.0でトレーニングせずにモデルを保存する

2020-02-14 python tensorflow keras tensorflow2.0

Tensorflow 2.0とKerasシーケンシャルAPIを使用してモデルを構築します。ベースライン比較のために、トレーニングされていないモデルを保存したいと思います。それ、どうやったら出来るの? epochs=0period=0を設定しようとしましたが、うまくいきませんでした。

ここに私のコードがあります:

network = NeuralNetwork() # sequential neural network
cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath='./models/', save_weights_only=False, period=0)
network.fit(x_train, y_train, epochs=0, callbacks=[cp_callback])

トレーニングされていないモデルを保存するにはどうすればよいですか?

Answers

モデルのインスタンス化後にmodel.saveを呼び出すだけです。

network = NeuralNetwork() # sequential neural network
network.save("/your/path/.h5") 

ModelCheckpointはコールバックとしてのみ呼び出すことができるため、トレーニング中に。

Related