simple-vgg16-cu

Simple VGG16 implemented in CUDA

Stars
6

Simple VGG16 in CUDA

This is a simple VGG16 net implemented in CUDA.

Simply run make; ./a.out to test the code. You may need to setup the environment for CUDA, cudnn, cublas.

The convolution layers use cudnn, and the FC layer uses cublas.

Related Projects