Train a Deep Graph Network - Amazon SageMaker