From 607863f3fab644a275c11886f74cd32d7048bc6c Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Wed, 17 Mar 2021 11:22:02 -0700
Subject: [PATCH] fix?

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

diff --git a/core/leras/layers/MsSsim.py b/core/leras/layers/MsSsim.py
index 6c8e931..cb48294 100644
--- a/core/leras/layers/MsSsim.py
+++ b/core/leras/layers/MsSsim.py
@@ -24,6 +24,6 @@ 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)
         # ssim_multiscale returns values in range [0, 1] (where 1 is completely identical)
         # subtract from 1 to get loss
-        return 1 - ms_ssim_val
+        return 1.0 - ms_ssim_val
 
 nn.MsSsim = MsSsim