模型参数的初始化

二次信任 提交于 2019-12-01 10:27:07

1.  tf.global_variables_initializer() 可以初始化所有变量。

import tensorflow as tfa=tf.Variable(tf.ones((2,3)),name='a')b=tf.Variable(tf.random_normal(shape=(2,3),stddev=0.35),name='b')sess=tf.Session()sess.run(tf.global_variables_initializer())print(sess.run(a))print(sess.run(b))

 

 

2.  tf.variables_initializer([var_list]) 仅初始化列表var_list种的值。

import tensorflow as tfa=tf.Variable(tf.ones((2,3)),name='a')b=tf.Variable(tf.random_normal(shape=(2,3),stddev=0.35),name='b')sess=tf.Session()# sess.run(tf.variables_initializer([a])) # 只初始化变量a,而变量b未作初始化,因此执行会报错。sess.run(tf.variables_initializer([a,b])) # 对变量a与b做了初始化,执行将不会报错print(sess.run(a))print(sess.run(b))

报错结果:

 

 正确结果:

 

 

3.  变量重复赋值并未报错,其结果如下:

此点证实模型参数可以用assign分配,可用checkpoint模型文件保存值赋值到变量种。

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!