Skip to content

Commit a70288d

Browse files
authored
Fix concat (#4755)
* fix init * update * fix cgan
1 parent 131a315 commit a70288d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

PaddleCV/gan/network/CGAN_network.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def __init__(self, batch_size=1):
4242

4343
def network_G(self, input, label, name="generator"):
4444
# concat noise and label
45-
y = fluid.layers.reshape(label, shape=[-1, self.y_dim, 1, 1])
45+
y = fluid.layers.reshape(label, shape=[-1, self.y_dim])
4646
xy = fluid.layers.concat([input, y], 1)
4747
o_l1 = linear(
4848
input=xy,

0 commit comments

Comments
 (0)