保存tf.layers.dense()网络的权重weights及失败原因

最新推荐文章于 2022-03-16 16:36:53 发布

onewarmheart

最新推荐文章于 2022-03-16 16:36:53 发布

阅读量4.2k

点赞数 5

分类专栏： Tensorflow用法文章标签： Tensorflow 深度学习

本文链接：https://blog.csdn.net/cminusminus/article/details/80872838

版权

在使用Tensorflow的tf.layers.dense构建网络时，为查看action网络权重变化，尝试将其保存。有两种方法：1) 保存为张量Tensor，通过layer的命名路径如'Actor_1/a/kernel:0'获取，可能遇到命名冲突问题；2) 保存为变量Variable，利用tf.get_variable直接取用。第二种方法便于将权重变化可视化。参考了StackOverflow和知乎的相关解答。

摘要由CSDN通过智能技术生成

这是我的第一篇CSDN博客：）

最近在写DDPG的时候发现action网络输出有问题，action网络输出层建立如下：

            self.a_net = tf.layers.dense(self.net, self.a_dim, 
                                activation=tf.nn.sigmoid,     
                                name='a',  
                                kernel_initializer=tf.truncated_normal_initializer(0,0.1),
                                bias_initializer=tf.constant_initializer(0.1),
                                trainable=trainable)

师兄建议把网络权重调出来看一下变化趋势，而我的action网络是用tf.layers.dense()构建的，所以查了一下如何保存tf.layers.dense()的网络的权重。

有如下两种方式：