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

update default kernel size

Browse source
This commit is contained in:
Jeremy Hummel 2019-10-20 00:38:55 -04:00
commit 87926317c7
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

@ -349,7 +349,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
class MsSSIM(object): class MsSSIM(object):
_MSSSIM_WEIGHTS = (0.0448, 0.2856, 0.3001, 0.2363, 0.1333) _MSSSIM_WEIGHTS = (0.0448, 0.2856, 0.3001, 0.2363, 0.1333)
def __init__(self, kernel_size=11, k1=0.01, k2=0.03, max_value=1.0, power_factors=_MSSSIM_WEIGHTS): def __init__(self, kernel_size=7, k1=0.01, k2=0.03, max_value=1.0, power_factors=_MSSSIM_WEIGHTS):
self.kernel_size = kernel_size self.kernel_size = kernel_size
self.k1 = k1 self.k1 = k1
self.k2 = k2 self.k2 = k2