I am Able to train my modelusing Sagemaker TensorFlow container.
Below is the code
model_dir = \'/opt/ml/model\' trai