Training a deep learning model using Amazon SageMaker