Fix a bug

Author: lizekai-richard

dd_ranking/metrics/dd_ranking_obj.py

@@ -195,7 +195,6 @@ def compute_hard_label_metrics(self, model, images, lr, hard_labels):
                 metric = validate(
                     model=model, 
                     loader=self.test_loader,
-                    aug_func=self.aug_func,
                     device=self.device
                 )
                 if metric['top1'] > best_acc1:
@@ -240,7 +239,6 @@ def compute_soft_label_metrics(self, model, images, lr, soft_labels):
                 metric = validate(
                     model=model, 
                     loader=self.test_loader,
-                    aug_func=self.aug_func,
                     device=self.device
                 )
                 if metric['top1'] > best_acc1:
@@ -500,7 +498,6 @@ def compute_hard_label_metrics(self, model, images, lr, hard_labels):
                 metric = validate(
                     model=model, 
                     loader=self.test_loader,
-                    aug_func=self.aug_func,
                     device=self.device
                 )
                 if metric['top1'] > best_acc1:

dd_ranking/utils/utils.py

@@ -524,7 +524,6 @@ def _backward(_loss):
 def validate(
     model,
     loader,
-    aug_func=None,
     device=torch.device('cuda'),
     logging=False,
     log_interval=10
@@ -545,7 +544,7 @@ def validate(
             last_batch = batch_idx == last_idx
             input = input.to(device)
             target = target.to(device)
-            input = aug_func(input)
+
             output = model(input)
             if isinstance(output, (tuple, list)):
                 output = output[0]