-
Book Overview & Buying
-
Table Of Contents
Accelerate Deep Learning Workloads with Amazon SageMaker
By :
Amazon SageMaker supports many popular ML and DL frameworks. Framework support in SageMaker is achieved using prebuilt Docker containers for inference and training tasks. Prebuilt SageMaker containers provide a great deal of functionality, and they allow you to implement a wide range of use cases with minimal coding. There are also real-life scenarios where you need to have a custom, runtime environment for training and/or inference tasks. To address these cases, SageMaker provides a flexible Bring-Your-Own (BYO) container feature.
In this chapter, we will review key supported DL frameworks and corresponding container images. Then, we will focus our attention on the two most popular DL frameworks, TensorFlow and PyTorch, and learn how to use them in Amazon SageMaker. Additionally, we will review a higher-level, state-of-the-art framework, Hugging Face, for NLP tasks, and its implementation for Amazon SageMaker.
Then, we will...