运行SGD优化器以构建CNN时出错



我在运行SGD优化器以构建CNN时出现以下错误

-->146优化器=SGD(0.01、0.8、0.0005,nesterov=True(147模型.编译(优化器=优化器,损失="类别交叉熵",度量=["准确性"](148#列车型号

类型错误:init((为参数"nesterov"获取了多个值

keras SGD优化器有3个参数,它接受kwargs,因此最好为参数提供密钥。

tf.keras.optimizers.SGD(learning_rate=0.01, momentum=0.0, nesterov=False, name="SGD")

如果更改代码后问题仍未解决。你必须升级Keras并以这种方式使用

optimizer = SGD(lr=0.01, momentum=0.8, decay=0.0005, nesterov=True)
model.compile(optimizer=optimizer, loss='categorical_crossentropy', metrics=['accuracy']) 

最新更新