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

fix: typo in new ms_ssim opt function

Browse source
This commit is contained in:
jh 2021-03-12 08:11:21 -08:00
commit 039559e4fd
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/leras/ops/__init__.py
View file

@ -333,7 +333,7 @@ def ms_ssim(img1, img2, resolution, kernel_size=11, k1=0.01, k2=0.03, max_value=
loss = tf.cast(loss, img_dtype)
return loss
nn.dssim = ms_ssim
nn.ms_ssim = ms_ssim
def space_to_depth(x, size):