python - 在使用 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])

如何保存未经训练的模型？

标签： pythontensorflowkerastensorflow2.0

Answer 1

您只需在模型实例化后调用 model.save ：

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

ModelCheckpoint 只能作为回调调用，所以在训练期间。