# AlexNet

AlexNet Inference on CUDANet

## Usage

1. Export pytorch AlexNet weight pretrained on ImageNet (requires pytorch and torchvision):

```sh
python alexnet.py
```

2. Follow the instructions from repository root to build CUDANet library.

3. Build AlexNet

```sh
mkdir build
cd build
cmake -S ..
make
```

4. Run AlexNet inference

```sh
alexnet ../alexnet_weights.bin ../image.jpg
```