Skip to content

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

fix bug in Conv2DTranspose For TF1.13 (introduced in 152c4c2c)

parent 55baf374
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/models/conv2d.py
View file @ 3862b72c
...@@ -218,7 +218,7 @@ def Conv2DTranspose( ...@@ -218,7 +218,7 @@ def Conv2DTranspose(
[shape_dyn[0], [shape_dyn[0],
shape_dyn[1] * strides2d[0], shape_dyn[1] * strides2d[0],
shape_dyn[2] * strides2d[1], shape_dyn[2] * strides2d[1],
channels_in]) filters])
out_shape3_sta = [None if inputs.shape[1] is None else inputs.shape[1] * strides2d[0], out_shape3_sta = [None if inputs.shape[1] is None else inputs.shape[1] * strides2d[0],
None if inputs.shape[2] is None else inputs.shape[2] * strides2d[1], None if inputs.shape[2] is None else inputs.shape[2] * strides2d[1],
filters] filters]
......
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