From bde700243cd5677f3bc1fb39ccd75fae9f3416ef Mon Sep 17 00:00:00 2001 From: iperov Date: Sun, 5 May 2019 19:22:26 +0400 Subject: [PATCH] fix for plaidML --- nnlib/nnlib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nnlib/nnlib.py b/nnlib/nnlib.py index 8f4d5ab..7f9be2d 100644 --- a/nnlib/nnlib.py +++ b/nnlib/nnlib.py @@ -315,7 +315,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator cs = (num1 - num0 + c2) / (den1 - den0 + c2) ssim_val = K.mean(luminance * cs, axis=(-3, -2) ) - return K.mean( (1.0 - ssim_val ) / 2.0 ) + return(1.0 - ssim_val ) / 2.0 return func