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