github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 22:34:25 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix normalization

Browse source
This commit is contained in:
Jeremy Hummel 2019-10-20 00:57:46 -04:00
commit 07a3e7964f
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
nnlib/nnlib.py
View file

@ -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)