diff --git a/PaddleCV/gan/network/Pix2pix_network.py b/PaddleCV/gan/network/Pix2pix_network.py index 3b37d821c8615eb4d5d2cbb76d4936d9ad02d273..b86806ba2237c88aa5c72e48fde08bde4179c7f1 100644 --- a/PaddleCV/gan/network/Pix2pix_network.py +++ b/PaddleCV/gan/network/Pix2pix_network.py @@ -335,7 +335,7 @@ def Unet_block(inputunet, use_bias=use_bias) if use_dropout: - upnorm = fluid.layers.dropout(upnorm, dropout_prob=0.5) + updeconv = fluid.layers.dropout(updeconv, dropout_prob=0.5) return fluid.layers.concat([inputunet, updeconv], 1)