search

ssnl / dataset-distillation

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

How to use it for custom model and dataset? #39

Closed Ebjderman closed 3 years ago

Ebjderman commented 3 years ago

Thank you for your nice work. Could you please provide an example about how to use it for distilling any custom dataset and employ user-defined network structure?

ssnl commented 3 years ago

It requires

  1. modifying https://github.com/SsnL/dataset-distillation/blob/master/datasets/__init__.py to include your dataset
  2. modifying https://github.com/SsnL/dataset-distillation/blob/master/networks/networks.py to include your network
  3. add proper argparsing for building dataset and network

The linked files have a couple examples you can probably follow.

Firegreat123 commented 2 years ago