From 12e819abb9f711db2076ea5d7e8e2d94edcbf625 Mon Sep 17 00:00:00 2001 From: lijianshe02 Date: Wed, 14 Oct 2020 11:21:13 +0000 Subject: [PATCH] refine psgan code according tho the reviewer's comment --- ppgan/models/discriminators/nlayers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppgan/models/discriminators/nlayers.py b/ppgan/models/discriminators/nlayers.py index 741233e..100af24 100644 --- a/ppgan/models/discriminators/nlayers.py +++ b/ppgan/models/discriminators/nlayers.py @@ -34,7 +34,7 @@ class NLayerDiscriminator(nn.Layer): input_nc (int) -- the number of channels in input images ndf (int) -- the number of filters in the last conv layer n_layers (int) -- the number of conv layers in the discriminator - norm_layer -- normalization layer + norm_type (str) -- normalization layer type """ super(NLayerDiscriminator, self).__init__() norm_layer = build_norm_layer(norm_type) -- GitLab