fixbug: metric_data always initialized

Author: xinntao

basicsr/models/sr_model.py

@@ -138,7 +138,7 @@ def nondist_validation(self, dataloader, current_iter, tb_logger, save_img):
         with_metrics = self.opt['val'].get('metrics') is not None
         if with_metrics:
             self.metric_results = {metric: 0 for metric in self.opt['val']['metrics'].keys()}
-            metric_data = dict()
+        metric_data = dict()
         pbar = tqdm(total=len(dataloader), unit='image')
 
         for idx, val_data in enumerate(dataloader):

basicsr/models/video_base_model.py

@@ -34,7 +34,7 @@ def dist_validation(self, dataloader, current_iter, tb_logger, save_img):
         if with_metrics:
             for _, tensor in self.metric_results.items():
                 tensor.zero_()
-            metric_data = dict()
+        metric_data = dict()
         # record all frames (border and center frames)
         if rank == 0:
             pbar = tqdm(total=len(dataset), unit='frame')

basicsr/models/video_recurrent_model.py

@@ -83,7 +83,7 @@ def dist_validation(self, dataloader, current_iter, tb_logger, save_img):
         if with_metrics:
             for _, tensor in self.metric_results.items():
                 tensor.zero_()
-            metric_data = dict()
+        metric_data = dict()
         num_folders = len(dataset)
         num_pad = (world_size - (num_folders % world_size)) % world_size
         if rank == 0: