tf.keras.models.save_model和优化器警告

Question

我使用以下方法创建了一个顺序模型tf.keras：model = tf.keras.Sequential()model.add(tf.keras.layers.Dense(8, input_dim=4))model.add(tf.keras.layers.Dense(3, activation=tf.nn.softmax))opt = tf.train.AdamOptimizer(learning_rate=0.001)model.compile(optimizer=opt, loss="categorical_crossentropy", metrics=["accuracy"])model.summary()之后，我使用train_on_batch以下代码创建了一个培训过程：EPOCHS=50for epoch in range(EPOCHS):  for metrics, labels in dataset:    # Calculate training loss and accuracy    tr_loss, tr_accuracy = model.train_on_batch(metrics, labels)当我尝试保存模型时，会收到警告。我不明白为什么，因为我将优化器作为的一部分包括在内model.compile：tf.keras.models.save_model(    model,    "./model/iris_model.h5",    overwrite=True,    include_optimizer=True)警告：tensorflow：TensorFlow优化器无法在实例化后访问优化器属性或优化器状态。因此，我们无法将优化器保存为模型保存文件的一部分。加载模型后，您将不得不再次编译模型。最好改用Keras优化器（请参阅keras.io/optimizers）。我使用的TF版本是1.9.0-rc2。

Answer 1

像这样的优化程序tf.keras.optimizers.Adam()将被保存，而tf.train.AdamOptimizer()不会保存在上model.save()。

在撰写本文时，一些关于TensorFlow的官方教程都使用了tf.train.*优化器，而我坚信选择这样做tf.keras.optimizers.*是最好的方法。