Fix a bug

Author: lizekai-richard

dd_ranking/metrics/soft_label.py

@@ -215,7 +215,7 @@ def compute_hard_label_metrics(self, model, image_tensor, image_path, lr, hard_l
                 loader=train_loader, 
                 loss_fn=loss_fn, 
                 optimizer=optimizer,
-                aug_func=self.aug_func if self.use_aug_for_hard else None,
+                aug_func=self.aug_func,
                 lr_scheduler=lr_scheduler, 
                 tea_model=self.teacher_model, 
                 device=self.device