由于在新版本的keras中(指大于等于2.3.0版本),将原来的 lr 这一参数重命名为 leraning_rate,因此将 leraning_rate 重新改为 lr 即可解决报错问题

# keras=2.2.4 写法 
model.compile(loss='categorical_crossentropy',
              optimizer=Adam(  lr  =lr_schedule(0)),
              metrics=['accuracy'])

# keras=2.3.0 写法 
model.compile(loss='categorical_crossentropy',
              optimizer=Adam(  learning_rate  =lr_schedule(0)),
              metrics=['accuracy'])


Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐