无需使用 Keras 在 Tensorflow 2.0 中训练即可保存模型
Save model without training in Tensorflow 2.0 with Keras
我使用 Tensorflow 2.0 和 Keras sequential 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])
如何保存未经训练的模型？
您只需在模型实例化后调用 model.save 即可：
network = NeuralNetwork() # sequential neural network
network.save("/your/path/.h5")
ModelCheckpoint 只能作为回调调用，因此在训练期间。
我使用 Tensorflow 2.0 和 Keras sequential 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])
如何保存未经训练的模型？
您只需在模型实例化后调用 model.save 即可：
network = NeuralNetwork() # sequential neural network
network.save("/your/path/.h5")
ModelCheckpoint 只能作为回调调用，因此在训练期间。