Save model locally instead of S3

Author: mvsusp

tests/integ/test_tf.py

@@ -38,6 +38,7 @@ def tf_training_job(sagemaker_session, tf_full_version):
                                framework_version=tf_full_version,
                                training_steps=1,
                                evaluation_steps=1,
+                               checkpoint_path='/opt/ml/model',
                                hyperparameters={'input_tensor_name': 'inputs'},
                                train_instance_count=1,
                                train_instance_type='ml.c4.xlarge',
@@ -114,6 +115,7 @@ def test_tf_async(sagemaker_session):
                                role='SageMakerRole',
                                training_steps=1,
                                evaluation_steps=1,
+                               checkpoint_path='/opt/ml/model',
                                hyperparameters={'input_tensor_name': 'inputs'},
                                train_instance_count=1,
                                train_instance_type='ml.c4.xlarge',
@@ -156,6 +158,7 @@ def test_tf_vpc_multi(sagemaker_session, tf_full_version):
                            framework_version=tf_full_version,
                            training_steps=1,
                            evaluation_steps=1,
+                           checkpoint_path='/opt/ml/model',
                            hyperparameters={'input_tensor_name': 'inputs'},
                            train_instance_count=instance_count,
                            train_instance_type=instance_type,