Skip to content

Latest commit

 

History

History
 
 

alexnet

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

alexnet

AlexNet model architecture from the "One weird trick..." https://arxiv.org/abs/1404.5997`_ paper.

For the details, you can refer to pytorchx/alexnet

This alexnet is just several conv-relu-pool blocks followed by several fc-relu, nothing special. All layers can be implemented by tensorrt api, including addConvolution, addActivation, addPooling, addFullyConnected.

// 1. generate alexnet.wts from [pytorchx/alexnet](https://github.com/wang-xinyu/pytorchx/tree/master/alexnet)

// 2. put alexnet.wts into tensorrtx/alexnet

// 3. build and run

cd tensorrtx/alexnet

mkdir build

cd build

cmake ..

make

sudo ./alexnet -s   // serialize model to plan file i.e. 'alexnet.engine'

sudo ./alexnet -d   // deserialize plan file and run inference

// 4. see if the output is same as pytorchx/alexnet