Skip to content

S
seminar-breakout
Commit ace39425 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix #721

parent a8e47470
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
tensorpack/models/conv2d.py
View file @ ace39425
...@@ -167,11 +167,12 @@ def Conv2DTranspose( ...@@ -167,11 +167,12 @@ def Conv2DTranspose(
bias_regularizer=bias_regularizer, bias_regularizer=bias_regularizer,
activity_regularizer=activity_regularizer) activity_regularizer=activity_regularizer)
ret = layer.apply(inputs, scope=tf.get_variable_scope()) ret = layer.apply(inputs, scope=tf.get_variable_scope())
ret = tf.identity(ret, name='output')
ret.variables = VariableHolder(W=layer.kernel) ret.variables = VariableHolder(W=layer.kernel)
if use_bias: if use_bias:
ret.variables.b = layer.bias ret.variables.b = layer.bias
return tf.identity(ret, name='output') return ret
Deconv2D = Conv2DTranspose Deconv2D = Conv2DTranspose
tensorpack/models/fc.py
View file @ ace39425
...@@ -49,8 +49,9 @@ def FullyConnected( ...@@ -49,8 +49,9 @@ def FullyConnected(
bias_regularizer=bias_regularizer, bias_regularizer=bias_regularizer,
activity_regularizer=activity_regularizer) activity_regularizer=activity_regularizer)
ret = layer.apply(inputs, scope=tf.get_variable_scope()) ret = layer.apply(inputs, scope=tf.get_variable_scope())
ret = tf.identity(ret, name='output')
ret.variables = VariableHolder(W=layer.kernel) ret.variables = VariableHolder(W=layer.kernel)
if use_bias: if use_bias:
ret.variables.b = layer.bias ret.variables.b = layer.bias
return tf.identity(ret, name='output') return ret
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment