mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-08-22 22:34:25 -07:00
get rid of print statement
This commit is contained in:
parent
07a3e7964f
commit
a884c91d64
1 changed files with 1 additions and 2 deletions
|
@ -360,11 +360,10 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
|
|||
|
||||
# restrict mssim factors to those greater/equal to kernel size
|
||||
power_factors = [power_factors[i] for i in range(len(power_factors)) if resolution//(2**i) >= kernel_size]
|
||||
# normalize power factors
|
||||
# normalize power factors if reduced because of size
|
||||
if sum(power_factors) < 1.0:
|
||||
power_factors = [x/sum(power_factors) for x in power_factors]
|
||||
self.power_factors = power_factors
|
||||
print('power_factors', self.power_factors)
|
||||
|
||||
def __call__(self, y_true, y_pred):
|
||||
if nnlib.tf is not None:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue