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。

你可能感兴趣的:(Tensorflow,Tensorflow使用技巧)