fixed syntax bug in normalization, as suggested by @cclauss

NVlabs · Apr 14, 2019 · 6d3aa2b · 6d3aa2b
1 parent 44f15f1
commit 6d3aa2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/networks/normalization.py b/models/networks/normalization.py
@@ -77,7 +77,7 @@ def __init__(self, config_text, norm_nc, label_nc):
         elif param_free_norm_type == 'syncbatch':
             self.param_free_norm = SynchronizedBatchNorm2d(norm_nc, affine=False)
         elif param_free_norm_type == 'batch':
-            self.param_free_norm = BatchNorm2d(norm_nc, affine=False)
+            self.param_free_norm = nn.BatchNorm2d(norm_nc, affine=False)
         else:
             raise ValueError('%s is not a recognized param-free norm type in SPADE'
                              % param_free_norm_type)