From 07a3e7964f4a464a0be45ab4c690b50c8ea71970 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Sun, 20 Oct 2019 00:57:46 -0400 Subject: [PATCH] fix normalization --- nnlib/nnlib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nnlib/nnlib.py b/nnlib/nnlib.py index febe87b..e3983e3 100644 --- a/nnlib/nnlib.py +++ b/nnlib/nnlib.py @@ -362,7 +362,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator power_factors = [power_factors[i] for i in range(len(power_factors)) if resolution//(2**i) >= kernel_size] # normalize power factors if sum(power_factors) < 1.0: - power_factors *= 1.0/sum(power_factors) + power_factors = [x/sum(power_factors) for x in power_factors] self.power_factors = power_factors print('power_factors', self.power_factors)