首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何实现tensorflow cosine_decay

如何实现tensorflow cosine_decay
EN

Stack Overflow用户
提问于 2020-11-12 01:35:05
回答 1查看 285关注 0票数 0

当我在tensorflow中调用cosine_decay函数时,它显示以下错误:

'<' not supported between instances of 'CosineDecay' and 'int'

下面是我的代码:

代码语言:javascript
复制
  decay_steps = 1000
  lr_decayed_fn = tf.keras.experimental.CosineDecay(initial_learning_rate=0.01, decay_steps=1000)

  model.compile(optimizer=Adam(lr=lr_decayed_fn), loss=dice_coef_loss, metrics=[dice_coef])

我只是遵循了tensorflow上的教程,我不知道为什么会出现这个错误

EN

回答 1

Stack Overflow用户

发布于 2021-01-16 04:07:26

Adam(lr=lr_decayed_fn)更改为Adam(learning_rate=lr_decayed_fn)

tensorflow v2中的Adam优化器调用需要拼写出learning_rate,它没有将参数作为“lr”。请参阅此问题:https://github.com/tensorflow/tensorflow/issues/44172

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/64791367

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档