tensorflow重复使用共享变量
tensorflow
一个面向所有人的开源机器学习框架
项目地址:https://gitcode.com/gh_mirrors/te/tensorflow
免费下载资源
·
在构造CNN以及LSTM等模型的时候,需要重复使用共享权重。但是tensorflow默认是不允许重复使用变量的,会报一些InvalidArgumentError或者Attemping to use uninitialized value等错误。解决方法是使用variable_scope,并且设置该variable_scope的reuse=True。设置方法由两个:
with tf.variable_scope(scope_name,reuse=True):
tf.get_variable_scope().reuse_variables()
详情请参考Sharing Variables。
GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2
PiperOrigin-RevId: 663726708
2 个月前
91dac11a
This test overrides disabled_backends, dropping the default
value in the process.
PiperOrigin-RevId: 663711155
2 个月前
更多推荐
已为社区贡献10条内容
所有评论(0)