Dataset Condensation with Contrastive Signals

This repository is the official implementation of Dataset Condensation with Contrastive Signals (DCC).
The implementation is based on (https://github.com/VICO-UoE/DatasetCondensation).

Prerequisites

• pytorch (1.2.0)
• numpy (1.15.1)
• torchvision (0.4.0)
• scipy (1.1.0)

Training and evaluation

To train the DCC (or DSAC) model in the paper, run this command:

python main.py --ipc <1, 10, or 50> --model ConvNet --dataset 
   
     (--imagenet_group 
    
     ) --method 
     
       --contrast --save_path 
      

      
     
    
   

Please download ImageNet32x32 at (https://image-net.org/download-images)