alexnet-pytorch
AlexNet architecture implemented in pytorch.
This is based on original paper : http://www.cs.toronto.edu/~hinton/absps/imagenet.pdf
Get started :
python version > 3
pip install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
Credits,
Alex Krizhevsky and team ; dansuh17