ssnl / dataset-distillation

Open-source code for paper "Dataset Distillation"
https://ssnl.github.io/dataset_distillation
MIT License
778 stars 115 forks source link

Getting distilled images and testing on them #24

Closed elena-orlova closed 5 years ago

elena-orlova commented 5 years ago

Hello!

I'm wondering what is the correct way to get distilled images and test performance on them as well check performance on a normal dataset after training on the distilled images. I'm confused since there are many parameters and I've already read the advanced docs. So, to get distilled data, for example on Cifar10, I need to run

python main.py --mode distill_basic --dataset Cifar10 --arch AlexCifarNet  --distill_lr 0.001

Then distilled images are in file result.pth.

So, to train the network on usual full dataset I need to set --mode train, and if I want to test network performance after training on distilled data I need to set --mode train --phase test ?

Or, in other words, how to get results like in your paper where it’s said that you get 80% when fully trained against 54% with distilled data on CIFAR10?

Looking forward for your response! Thanks!