Update msssim_l1.py

willxxy · Feb 29, 2024 · 667d89c · 667d89c
1 parent 0c4749a
commit 667d89c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/msssim_l1.py b/msssim_l1.py
@@ -1,9 +1,10 @@
 class MS_SSIM_L1_LOSS(nn.Module):
     """
     Created on Thu Dec  3 00:28:15 2020
-
+    Modified on Wed Feb 38 20:19:10 2024
+    
     @orig-author for 3C: Yunpeng Li, Tianjin University
-    @orig-author for 1C: William Jongwon Han, Carnegie Mellon University
+    @modified-author for 1C: William Jongwon Han, Carnegie Mellon University
     """
     # Have to use cuda, otherwise the speed is too slow.
     def __init__(self, gaussian_sigmas=[0.5, 1.0, 2.0, 4.0, 8.0],