fix a bug: change l2 to mse (#570)

* add calc_psnr script * delete calc_psnr script * change l2 to mse --------- Co-authored-by: Xintao <[email protected]>
XPixelGroup · Feb 2, 2023 · 5521283 · 5521283
1 parent f6b3790
commit 5521283
Showing 1 changed file with 1 addition and 1 deletion.
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: