If you are just prototyping you might be able to just use a pretrained torchvision model:
https://pytorch.org/vision/stable/models.html 5
Downloading from the original website 1 is often tricky, but the academic torrents 2 version works well if you have a way to mount or access > 150GB of storage.