# DarKnight: A Data Privacy Scheme for Training andInference of Deep Neural Networks

## GPU Inference & Training Simulation
### Requirements
To install requirements:

```setup
 $ pip install -r requirements.txt
```
> Be sure to modify paths to actual datasets in `dataset.py` so that dataloaders can load images correctly.

### DarKnight Inference on ImageNet
To run DarKnight inference on ImageNet run the command below:
``` Inference
 $ ./run_imageNet_acc.sh
```
> The script above will run DarKnight inference over ImageNet validation set using `vgg16`, `vgg19` and `mobilenet`.
### DarKnight Training
To run DarKnight training run the command below:
``` Training
 $ ./run.sh [model name] [dataset name]
```
> Current supported models are `vgg16`, `vgg19` and `mobilenet`. Current supported datasets are `cifar10` and `cifar100`.