diff --git a/README.md b/README.md index e04471ca8..669f5620f 100644 --- a/README.md +++ b/README.md @@ -119,3 +119,5 @@ If you have any questions, please email `xintao.alpha@gmail.com`, `xintao.wang@o

+ +![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 index d2e965526..c95b2b77c 100644 --- 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: