Fix a bug

Author: lizekai-richard

README.md

@@ -258,6 +258,7 @@ Please check out [CONTRIBUTING.md](./CONTRIBUTING.md) for how to get involved.
 - [Dai Liu](https://scholar.google.com/citations?user=3aWKpkQAAAAJ&hl=en)
 - [Ziheng Qin](https://henryqin1997.github.io/ziheng_qin/)
 - [Kaipeng Zhang](https://kpzhang93.github.io/)
+- [Zheng Zhu](http://www.zhengzhu.net/)
 - [Zhangyang Wang](https://vita-group.github.io/)
 - [Bo Zhao](https://www.bozhao.me/)
 - [Yang You](https://www.comp.nus.edu.sg/~youy/)

dd_ranking/metrics/soft_label.py

@@ -23,9 +23,9 @@ def __init__(self, config: Config=None, dataset: str='CIFAR10', real_data_path:
                  soft_label_criterion: str='kl', data_aug_func: str='cutmix', aug_params: dict={'beta': 1.0}, soft_label_mode: str='S',
                  optimizer: str='sgd', lr_scheduler: str='step', temperature: float=1.0, weight_decay: float=0.0005, 
                  momentum: float=0.9, num_eval: int=5, im_size: tuple=(32, 32), num_epochs: int=300, use_zca: bool=False,
-                 real_batch_size: int=256, syn_batch_size: int=256, default_lr: float=0.01, save_path: str=None, use_aug_for_hard: bool=False,
-                 stu_use_torchvision: bool=False, tea_use_torchvision: bool=False, num_workers: int=4, teacher_dir: str='./teacher_models', 
-                 custom_train_trans: transforms.Compose=None, custom_val_trans: transforms.Compose=None, device: str="cuda"):
+                 real_batch_size: int=256, syn_batch_size: int=256, default_lr: float=0.01, save_path: str=None, stu_use_torchvision: bool=False, 
+                 tea_use_torchvision: bool=False, num_workers: int=4, teacher_dir: str='./teacher_models', custom_train_trans: transforms.Compose=None, 
+                 custom_val_trans: transforms.Compose=None, device: str="cuda"):
 
         if config is not None:
             self.config = config
@@ -45,12 +45,14 @@ def __init__(self, config: Config=None, dataset: str='CIFAR10', real_data_path:
             num_eval = self.config.get('num_eval')
             im_size = self.config.get('im_size')
             num_epochs = self.config.get('num_epochs')
+            use_zca = self.config.get('use_zca')
             real_batch_size = self.config.get('real_batch_size')
             syn_batch_size = self.config.get('syn_batch_size')
             default_lr = self.config.get('default_lr')
             save_path = self.config.get('save_path')
             num_workers = self.config.get('num_workers')
-            use_torchvision = self.config.get('use_torchvision')
+            stu_use_torchvision = self.config.get('stu_use_torchvision')
+            tea_use_torchvision = self.config.get('tea_use_torchvision')
             custom_train_trans = self.config.get('custom_train_trans')
             custom_val_trans = self.config.get('custom_val_trans')
             teacher_dir = self.config.get('teacher_dir')
@@ -100,7 +102,6 @@ def __init__(self, config: Config=None, dataset: str='CIFAR10', real_data_path:
             self.aug_func = Cutmix(aug_params)
         else:
             self.aug_func = None
-        self.use_aug_for_hard = use_aug_for_hard
 
         if not save_path:
             save_path = f"./results/{dataset}/{model_name}/ipc{ipc}/obj_scores.csv"