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

fixes loss range

Browse source
This commit is contained in:
jh 2021-03-17 12:19:10 -07:00
commit 5a8d9f66f2
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/leras/layers/MsSsim.py
View file

@ -24,7 +24,7 @@ class MsSsim(nn.LayerBase):
ms_ssim_val = tf.image.ssim_multiscale(y_true_t, y_pred_t, max_val, power_factors=self.power_factors, filter_size=self.kernel_size) ms_ssim_val = tf.image.ssim_multiscale(y_true_t, y_pred_t, max_val, power_factors=self.power_factors, filter_size=self.kernel_size)
# ssim_multiscale returns values in range [0, 1] (where 1 is completely identical) # ssim_multiscale returns values in range [0, 1] (where 1 is completely identical)
# subtract from 1 to get loss # subtract from 1 to get loss
return (1.0 - ms_ssim_val) / 2.0 return 1.0 - ms_ssim_val
nn.MsSsim = MsSsim nn.MsSsim = MsSsim