tf.global_variables_initializer()与tf.local_variables_initializer()的区别

狼啸风云

修改于 2022-09-04 21:07:06

3.4K0

修改于 2022-09-04 21:07:06

文章被收录于专栏：计算机视觉理论及其实现计算机视觉理论及其实现

一、tf.global_variables_initializer()

tf.global_variables_initializer()添加节点用于初始化所有的变量(GraphKeys.VARIABLES)。返回一个初始化所有全局变量的操作（Op）。在你构建完整个模型并在会话中加载模型后，运行这个节点。

能够将所有的变量一步到位的初始化，非常的方便。通过feed_dict, 你也可以将指定的列表传递给它，只初始化列表中的变量。

示例代码如下：

sess.run(tf.global_variables_initializer(),
feed_dict={
        learning_rate_dis: learning_rate_val_dis,
        adam_beta1_d_tf: adam_beta1_d,
        learning_rate_proj: learning_rate_val_proj,
        lambda_ratio_tf: lambda_ratio,
        lambda_l2_tf: lambda_l2,
        lambda_latent_tf: lambda_latent,
        lambda_img_tf: lambda_img,
        lambda_de_tf: lambda_de,
        adam_beta1_g_tf: adam_beta1_g,
        })
# learning_rate_dis为设置的变量，learning_rate_val_dis为我设置的具体的值。后续同理

二、tf.local_variables_initializer()

tf.local_variables_initializer()返回一个初始化所有局部变量的操作（Op）。初始化局部变量（GraphKeys.LOCAL_VARIABLE）。GraphKeys.LOCAL_VARIABLE中的变量指的是被添加入图中，但是未被储存的变量。关于储存，请了解tf.train.Saver相关内容，在此处不详述，敬请原谅。

示例代码与上面的类似：

sess.run(tf.local_variables_initializer(),
feed_dict={
     learning_rate_dis: learning_rate_val_dis,
     adam_beta1_d_tf: adam_beta1_d,
     learning_rate_proj: learning_rate_val_proj,
     lambda_ratio_tf: lambda_ratio,
     lambda_l2_tf: lambda_l2,
     lambda_latent_tf: lambda_latent,
     lambda_img_tf: lambda_img,
     lambda_de_tf: lambda_de,
     adam_beta1_g_tf: adam_beta1_g,
     })
# learning_rate_dis为设置的变量，learning_rate_val_dis为我设置的具体的值。后续同理

转载地址：https://blog.csdn.net/yyhhlancelot/article/details/81415137

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2019年06月26日，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

编程算法

登录后参与评论

0 条评论

热度