Merge branch 'master' into feat/colorspace

XPixelGroup · Feb 2, 2023 · fd00a20 · fd00a20
2 parents 4013631 + 5521283
commit fd00a20
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -119,3 +119,5 @@ If you have any questions, please email `[email protected]`, `xintao.wang@o
   <img src="https://user-images.githubusercontent.com/17445847/134879983-6f2d663b-16e7-49f2-97e1-7c53c8a5f71a.jpg"  height="300">  &emsp;
   <img src="https://user-images.githubusercontent.com/17445847/139572512-8e192aac-00fa-432b-ac8e-a33026b019df.png"  height="300">
 </p>
+
+![visitors](https://visitor-badge.glitch.me/badge?page_id=XPixelGroup/BasicSR) (start from 2022-11-06)
diff --git a/basicsr/losses/basic_loss.py b/basicsr/losses/basic_loss.py
@@ -189,7 +189,7 @@ def __init__(self,
         if self.criterion_type == 'l1':
             self.criterion = torch.nn.L1Loss()
         elif self.criterion_type == 'l2':
-            self.criterion = torch.nn.L2loss()
+            self.criterion = torch.nn.MSELoss()
         elif self.criterion_type == 'fro':
             self.criterion = None
         else: