diff --git a/basicsr/models/sr_model.py b/basicsr/models/sr_model.py index b9d1a70f3..c493e6941 100644 --- a/basicsr/models/sr_model.py +++ b/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): diff --git a/basicsr/models/video_base_model.py b/basicsr/models/video_base_model.py index 326742736..20b0cc342 100644 --- a/basicsr/models/video_base_model.py +++ b/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') diff --git a/basicsr/models/video_recurrent_model.py b/basicsr/models/video_recurrent_model.py index 05ec495dc..07142dc0e 100644 --- a/basicsr/models/video_recurrent_model.py +++ b/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: