【Tensorflow】解决使用SGD优化器报错

使用tensorflow 2.0、keras2.8时,两个库若即若离,互相干扰,导致了很多报错。

在使用优化器SGD时,尝试几种方法。

from keras.optimizer_v1 import sgd


model.compile(loss="mse", optimizer=sgd(lr=1e-6, momentum=0.9))

报错

from tensorflow.python.keras.optimizer_v1 import sgd

model.compile(loss="mse", optimizer=sgd(lr=1e-6, momentum=0.9))

还是报错

import tensorflow


model.compile(loss="mse", optimizer=tf.keras.optimizers.SGD(lr=1e-6, momentum=0.9))

正确

虽然新版本keras移到了tensorflow.python下,但是SGD优化器好像没移过去,而RMSprop则使用新版本方法。

详情参考Tensorflow V2.80 SGD官方文档https://tensorflow.google.cn/api_docs/python/tf/keras/optimizers/SGD

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值