If you use Pytorch (https://pytorch.org) you can use the implemented models in torchvision (https://pytorch.org/docs/stable/torchvision/models.html):

import torchvision.models as models alexnet = models.alexnet()

Also, you use the parameter "pretrained =True" for a model pre-trained on ImageNet

For the data set you can use the implemented for Pytorch in https://pytorch.org/docs/stable/torchvision/datasets.html?highlight=dataset%20image#torchvision.datasets.ImageFolder.

Hi

I hope this will help u

https://github.com/topics/alexnet?l=jupyter+notebook

Regards

Have a look at https://www.pyimagesearch.com/2020/03/16/detecting-covid-19-in-x-ray-images-with-keras-tensorflow-and-deep-learning/