0

事情就是这样。

我建立了一个训练有素的网络并将 .meta 和 .data 保存在 ckpt 中。优化器是 MomentumOptimizer:

tf.train.MomentumOptimizer(learning_rate=lr, momentum=0.9, name='Momentum'),它也保存在 .meta 中。

但是我忘记保存与 Momentum 相关的参数,所以当我加载图形表单 .meta 时,会出现错误:

尝试使用未初始化的值 fc.bias/Momentum

我知道我可以使用 initializer() ,但是这样如何设置 momentum=0.9呢?这个初始化器()会只给一个随机值momentum吗?

我已经训练了很长时间的模型,我没有办法再次从头开始训练它来保存 xx/Momentum。那么我能做些什么来修复它呢?momentum=0.9也就是说,加载 .meta 时如何在 MomentumOptimizer 中设置 ?

4

1 回答 1

0

momentum=0.9保存模型时会保存,所以不用担心。或者您可以制作momentum一个占位符,以便在重新加载模型时加载您想要的任何内容。

于 2020-06-14T06:09:52.127 回答