Skip to content
Permalink
Browse files

fix for plaidML

  • Loading branch information...
iperov committed May 5, 2019
1 parent 35ab63f commit bde700243cd5677f3bc1fb39ccd75fae9f3416ef
Showing with 1 addition and 1 deletion.
  1. +1 −1 nnlib/nnlib.py
@@ -315,7 +315,7 @@ def reducer(x):
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

0 comments on commit bde7002

Please sign in to comment.
You can’t perform that action at this time.