Image Classification - PyTorch

This is a supervised image clasification algorithm which supports fine-tuning of many pre-trained models available in Pytorch Hub. The following sample notebook demonstrates how to use the Sagemaker Python SDK for Image Classification for using these algorithms.

For detailed documentation please refer Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK

Available Models

Model ID

Fine Tunable?

Latest Version

Min SDK Version

Source

pytorch-ic-alexnet

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-densenet121

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-densenet161

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-densenet169

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-densenet201

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-googlenet

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-mobilenet-v2

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-resnet101

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-resnet152

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-resnet18

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-resnet34

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-resnet50

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-resnext101-32x8d

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-resnext50-32x4d

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-shufflenet-v2-x1-0

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-squeezenet1-0

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-squeezenet1-1

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-vgg11

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-vgg11-bn

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-vgg13

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-vgg13-bn

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-vgg16

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-vgg16-bn

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-vgg19

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-vgg19-bn

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-wide-resnet101-2

True

2.2.4

2.75.0

Pytorch Hub

pytorch-ic-wide-resnet50-2

True

2.2.4

2.75.0

Pytorch Hub