Zero-shot_Knowledge_Distillation_Pytorch

KO | EN

gitlite — search

#python #swift #android #nodejs #node #python3 #machine-learning #ios #javascript #react #rust #security

Zero-shot_Knowledge_Distillation_Pytorch

★ 32 Open GitHub ↗

ZSKD with PyTorch

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Zero-Shot Knowledge Distillation in Deep Networks Pytorch

![Python version support](https://img.shields.io/badge/python-3.6-blue.svg)
![PyTorch version support](https://img.shields.io/badge/pytorch-1.7.0-red.svg)

:star: Star us on GitHub — it helps!!

PyTorch implementation for *[Zero-Shot Knowledge Distillation in Deep Networks](https://arxiv.org/abs/1905.08114)*

## Install

You will need a machine with a GPU and CUDA installed.  
Then, you prepare runtime environment:

```shell
   pip install -r requirements.txt
   ```

## Use

For mnist dataset,

```shell
python main.py --dataset=mnist --t_train=False --num_sample=12000 --batch_size=200 
```

For cifar10 dataset,

```shell
python main.py --dataset=cifar10 --t_train=False --num_sample=24000 --batch_size=100
```

Arguments:

- `dataset` - available dataset: ['mnist', 'cifar10', 'cifar100']
- `t_train` - Train teacher network?? 
	- if True, train teacher network
	- elif False, load trained teacher network
- `num_sample` - Number of DIs crafted per category
- `beta` - Beta scaling vectors
- `batch_size` - batch size
- `lr` - learning rate
- `iters` - iteration number
- `s_save_path` - save path for student network
- `do_genimgs` - generate synthesized images from ZSKD??
	- if True, generate images
	- elif False, you must have the synthesized images that are generated from ZSKD

## Result examples for MNIST dataset

![2](https://raw.githubusercontent.com/da2so/Zero-shot_Knowledge_Distillation_Pytorch/master/assets/fig1.png)

## Understanding this method(algorithm)

:white_check_mark: Check my blog!!
[Here](https://da2so.github.io/2020-08-12-Zero_Shot_Knowledge_Distillation_in_Deep_Networks/)

Zero-shot_Knowledge_Distillation_Pytorch

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README