Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 09:44:28 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
88e852a6e5addfb1cc4427c36aeaf7673f9b6ae3
CUDANet/examples/alexnet/README.md
LordMathis 157d4c736b Add Alexnet README
2024-04-22 21:59:08 +02:00

394 B
Raw Blame History

AlexNet

AlexNet Inference on CUDANet

Usage

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

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

  2. Build AlexNet

mkdir build
cd build
cmake -S ..
make
  1. Run AlexNet inference
alexnet ../alexnet_weights.bin ../image.jpg