Skip to content

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

fix unpooling with none shape

parent 4bd56d0f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tensorpack/models/pool.py
View file @ b2ac1dae
...@@ -73,8 +73,8 @@ def UnPooling2x2ZeroFilled(x): ...@@ -73,8 +73,8 @@ def UnPooling2x2ZeroFilled(x):
out_size = [-1, sh[1] * 2, sh[2] * 2, sh[3]] out_size = [-1, sh[1] * 2, sh[2] * 2, sh[3]]
return tf.reshape(out, out_size) return tf.reshape(out, out_size)
else: else:
sh = tf.shape(x) shv = tf.shape(x)
ret = tf.reshape(out, tf.pack([-1, sh[1] * 2, sh[2] * 2, sh[3]])) ret = tf.reshape(out, tf.pack([-1, shv[1] * 2, shv[2] * 2, sh[3]]))
ret.set_shape([None, None, None, sh[3]]) ret.set_shape([None, None, None, sh[3]])
return ret 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