Skip to content

S
seminar-breakout
Commit 05f7ba8f authored by Yuxin Wu's avatar Yuxin Wu Committed by GitHub
Browse files
Options

fix Deconv. fix #154

parent 136174c9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/models/conv2d.py
View file @ 05f7ba8f
...@@ -141,7 +141,7 @@ def Deconv2D(x, out_shape, kernel_shape, ...@@ -141,7 +141,7 @@ def Deconv2D(x, out_shape, kernel_shape,
for k in out_shape: for k in out_shape:
if not isinstance(k, int): if not isinstance(k, int):
raise ValueError("[Deconv2D] out_shape {} is invalid!".format(k)) raise ValueError("[Deconv2D] out_shape {} is invalid!".format(k))
out_channel = out_shape[channel_axis] out_channel = out_shape[channel_axis - 1]
shp3_static = shp3_dyn = out_shape shp3_static = shp3_dyn = out_shape
filter_shape = kernel_shape + [out_channel, in_channel] filter_shape = kernel_shape + [out_channel, in_channel]
......
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