使用 Keras 在 Tensorflow 2.0 中无需训练即可保存模型

Question

我使用 Tensorflow 2.0 和 Keras 顺序 API 来构建模型。 我想保存一个未经训练的模型用于基线比较。 我怎么做？ 我试图设置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 只能作为回调调用，所以在训练期间。