Keras 卷积层错误:未找到 DNN 库

问题描述 投票:0回答:1

我在执行 model.fit 期间遇到 Keras 卷积神经网络问题。错误信息如下: `未找到 DNN 库。 [[节点顺序/conv1d/Conv1D]] [操作:__inference_train_function_10911]

model.compile(loss="categorical_crossentropy", optimizer="adam", metrics=["accuracy"])
train_model(model, name_network)

# ...

history = model.fit(
    X_train,
    y_train,
    epochs=nb_epoch,
    batch_size=batch_size,
    validation_data=(X_test, y_test),
    verbose=1,
    callbacks=[time_callback],
)

# ...

`

任何有关如何解决此问题的见解或建议将不胜感激。预先感谢您的帮助!

python tensorflow keras deep-learning cudnn
1个回答
0
投票

您可以使用以下命令导入 DNN 库:

model = tf.keras.Sequential([
    tf.keras.layers.Dense(128, activation='relu', input_shape=(input_size,)),
])

你需要使用.Dense

© www.soinside.com 2019 - 2024. All rights reserved.