github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 13:32:09 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix for plaidML

Browse source
This commit is contained in:
iperov 2019-05-05 19:22:26 +04:00
parent 35ab63f752
commit bde700243c
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

@ -315,7 +315,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
cs = (num1 - num0 + c2) / (den1 - den0 + c2) cs = (num1 - num0 + c2) / (den1 - den0 + c2)
ssim_val = K.mean(luminance * cs, axis=(-3, -2) ) 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 return func