Tensorflow 自定义loss的情况下初始化部分变量方式

作者:I_will_____ 时间:2023-02-26 22:43:39 

一般情况下,tensorflow里面变量初始化过程为:


 #variables ...........
 #.....................
 init = tf.initialize_all_variables()
 sess.run(init)

这里 tf.initialize_all_variables() 会初始化所有的变量。

实际过程中,假设有a, b, c三个变量,其中a已经被初始化了,只想单独初始化b,c,那么:


 #variables ...
 ...
 init = tf.variables_initializer([b,c])
 sess.run(init)

此外,如果自行修改了optimizer,如下代码就会报错:


 #definition of variables a, b, c ...
 ....
 my_optimizer = tf.train.RMSProp(learning_rate = 0.1).minimize(my_cost)
 init = tf.variables_initializer([b,c])
 sess.run(init)

这是因为自己定义的optimizer会生成新的variables,但是在init里面并没有初始化,所以无法访问,会报错。解决方法如下:


 a = tf.Variables(...)      #line N
 temp = set(tf.all_variables())
 b = tf.Variables(...)
 c = tf.Variables(...)
 #definition of my optimizer
 optimizer = tf.train.......
 init = tf.variables_initializer(set(tf.all_varialbles())-temp) # line M
 sess.run(init)

首先,temp = set(tf.all_variables()) 将该行(line N)代码之前的所有变量保存在temp中,接下来定义变量b, c,以及自定义的optimizer,然后 set(tf.all_varialbles()存储了改行(line M)之前的所有变量(包括optimizer生成的变量以及temp中所含的变量),set(tf.all_varialbles())-temp相减得到line N~M这几行定义的变量。

来源:https://blog.csdn.net/chenxicx1992/article/details/56483180

标签:Tensorflow,loss,初始化,变量
0
投稿

猜你喜欢

  • python实现二维数组的对角线遍历

    2023-05-15 06:08:01
  • GoLang中拼接字符串性能优化方法详解

    2024-04-28 09:16:46
  • Python复制Word内容并使用格式设字体与大小实例代码

    2023-01-10 05:48:20
  • sqlserver 触发器学习(实现自动编号)

    2012-10-07 10:58:03
  • Python进阶:生成器 懒人版本的迭代器详解

    2021-12-21 16:08:05
  • 使用keras内置的模型进行图片预测实例

    2021-12-27 17:54:29
  • Python 中如何使用 virtualenv 管理虚拟环境

    2022-02-20 00:57:44
  • Python实现的视频播放器功能完整示例

    2023-05-04 06:33:56
  • python 一维二维插值实例

    2022-07-17 10:09:08
  • javascript阻止事件冒泡和浏览器的默认行为

    2007-12-28 13:13:00
  • python中matplotlib调整图例位置的方法实例

    2023-09-11 08:52:13
  • vue在页面中如何使用window全局变量

    2024-05-09 15:14:49
  • python实现计算器小功能

    2022-07-15 04:33:56
  • ASP访问带多个参数的存储过程

    2008-10-14 16:45:00
  • 解决sublime+python3无法输出中文的问题

    2023-09-20 16:26:20
  • 通过Pandas读取大文件的实例

    2023-12-25 21:18:31
  • Python NumPy实现数组排序与过滤示例分析讲解

    2021-07-20 08:33:20
  • asp如何将产生的密码记录并发送给用户?

    2009-11-26 20:50:00
  • asp如何用CDONTS发送带附件的邮件?

    2010-06-11 19:57:00
  • 在CentOS上MySQL数据库服务器配置方法

    2024-01-19 07:01:21
  • asp之家 网络编程 m.aspxhome.com