From 039559e4fdb8cddb1b02f048b53971b6470f0c18 Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Fri, 12 Mar 2021 08:11:21 -0800
Subject: [PATCH] fix: typo in new ms_ssim opt function

---
 core/leras/ops/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/leras/ops/__init__.py b/core/leras/ops/__init__.py
index af59c6c..b7f1bab 100644
--- a/core/leras/ops/__init__.py
+++ b/core/leras/ops/__init__.py
@@ -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):