Skip to content

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

fix static batch size in Conv2DTranspose

parent f7e35411
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/models/conv2d.py
View file @ eb25cd7f
......@@ -246,7 +246,7 @@ def Conv2DTranspose(
shape4d(strides, data_format=data_format),
padding=padding.upper(),
data_format=data_format)
conv.set_shape(tf.TensorShape([None] + out_shape3_sta))
conv.set_shape(tf.TensorShape([shape_sta[0]] + out_shape3_sta))
ret = tf.nn.bias_add(conv, b, data_format=data_format) if use_bias else conv
if activation is not None:
......
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