Keras 2.14 中的优化器格式发生变化,导致简单模型无法导入

作者: YusifCodes创建于 2023年11月3日更新于 2025年7月12日
标签DL4J Keras

嘿大家, 我在加载简单的 Python keras 模型时遇到了问题。Python keras 模型: `model = keras.Sequential([ keras.layers.Dense(32, activation='relu', input_shape=(132,)), keras.layers.Dropout(.2), keras.layers.Dense(16, activation='relu'), keras.layers.Dense(41, activation='softmax') ]) optimizer = SGD(learning_rate=0.1) model.compile(optimizer=optimizer, loss=keras.losses.sparse_categorical_crossentropy, metrics=['accuracy'])

model.fit(train_x, train_y, epochs=10, batch_size=32, validation_split=0.1)

test_loss, test_accuracy = model.evaluate(test_x, test_y)
print(f"Test accuracy: {test_accuracy}")

model.save("test_model.h5")`

Java 代码出现错误: MultiLayerNetwork model = KerasModelImport.importKerasSequentialModelAndWeights("path/to/model.h5");

错误本身: org.deeplearning4j.nn.modelimport.keras.exceptions.UnsupportedKerasConfigurationException 错误消息: 具有名称 Custom>SGD的优化器无法与 DL4J 优化器匹配。请注意,自定义 TFOptimizers 不受模型导入支持。

内容来源: deeplearning4j/deeplearning4j