Image Classification - PyTorch — sagemaker 2.199.0 documentation (original) (raw)

• »
• Built-in Algorithms »
• Vision »
• Image Classification - PyTorch
• Edit on GitHub

This is a supervised image clasification algorithm which supports fine-tuning of many pre-trained models available in Pytorch Hub. The followingsample notebookdemonstrates 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¶