From 87926317c743bd2f57e77f728154de56034bb3e3 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Sun, 20 Oct 2019 00:38:55 -0400 Subject: [PATCH] update default kernel size --- nnlib/nnlib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nnlib/nnlib.py b/nnlib/nnlib.py index 8107786..40851f2 100644 --- a/nnlib/nnlib.py +++ b/nnlib/nnlib.py @@ -349,7 +349,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator class MsSSIM(object): _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.k1 = k1 self.k2 = k2