最近学习了指数衰减学习率,想看一下自己的模型学习率是否设置成功,百度了好久没有找到怎么实时打印学习率,不得不吐槽一下国内学习平台还不太行啊。
然后翻了一下墙找到一个比较好的答案,可以做到。
首先定义一个类,把你的模型传进来,然后调用show方法就可以了。
Tensorflow2.x 实时打印学习率
最新推荐文章于 2024-06-19 16:39:26 发布
最近学习了指数衰减学习率,想看一下自己的模型学习率是否设置成功,百度了好久没有找到怎么实时打印学习率,不得不吐槽一下国内学习平台还不太行啊。
然后翻了一下墙找到一个比较好的答案,可以做到。
首先定义一个类,把你的模型传进来,然后调用show方法就可以了。