Сохранить модель без обучения в Tensorflow 2.0 с помощью Keras

Question

Я использую Tensorflow 2.0 и последовательный API Keras для построения модели. Я хотел бы сохранить неподготовленную модель для базового сравнения. Как я могу это сделать? Я попытался установить epochs=0 и period=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])

Как сохранить неподготовленную модель?

Answer 1

Вы просто должны вызвать model.save после создания экземпляра вашей модели:

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

ModelCheckpoint может быть вызван только как обратный вызов, поэтому во время обучения.