tensorflow 获取权重与可训练变量

hyrbngr7  于 2022-11-16  发布在  其他
关注(0)|答案(1)|浏览(166)

Tensorflow中的model.get_weights()model.trainable_variables似乎在不同的数据类型中返回相同的值。前者返回数组列表,后者返回Tensor数组。(如果我没有弄错的话)
请,解释在哪个上下文中更好地使用每一个?此外,我试图比较他们,但没有运气,如果可能的话,请帮助与比较代码。

  • 谢谢-谢谢
oiopk7p5

oiopk7p51#

模型.get_weights()函数:

  • 它以NumPy数组的形式返回层的当前权重。
  • 该函数返回包含与该层相关联的可训练权重值和不可训练权重值的NumPy数组的列表,该列表然后可用于将状态加载到类似参数化的层中。
    模型.可训练变量:
  • 它返回在一个层中trainable = True时创建的所有变量。默认情况下,它在所有层中设置为true。

因此,model.get_weights()model.trainable_variables返回相同的值。请参阅此gist以获得更好的理解。谢谢!

相关问题