def tensorflow_estimator(): s3_output_location = 's3://sagemaker/models' s3_source_location = 's3://sagemaker/source' estimator = TensorFlow( entry_point='tf_train.py', role=EXECUTION_ROLE, framework_version='1.13', instance_count=1, instance_type='ml.p2.xlarge', output_path=s3_output_location, source_dir=s3_source_location, image_uri=TENSORFLOW_IMAGE, model_dir=False, hyperparameters={ 'training_steps': 1000, 'evaluation_steps': 100, 'checkpoint_path': 's3://sagemaker/models/sagemaker-tensorflow/checkpoints', }) estimator.debugger_hook_config = DebuggerHookConfig( s3_output_path='s3://sagemaker/models/debug') estimator.sagemaker_session = MagicMock() estimator.sagemaker_session.boto_region_name = 'us-east-1' estimator.sagemaker_session._default_bucket = 'sagemaker' return estimator
def tensorflow_estimator(): s3_output_location = 's3://sagemaker/models' s3_source_location = 's3://sagemaker/source' estimator = TensorFlow( entry_point='tf_train.py', role=EXECUTION_ROLE, framework_version='1.13', training_steps=1000, evaluation_steps=100, train_instance_count=1, train_instance_type='ml.p2.xlarge', output_path=s3_output_location, source_dir=s3_source_location, image_name=TENSORFLOW_IMAGE, checkpoint_path='s3://sagemaker/models/sagemaker-tensorflow/checkpoints' ) estimator.sagemaker_session = MagicMock() estimator.sagemaker_session.boto_region_name = 'us-east-1' estimator.sagemaker_session._default_bucket = 'sagemaker' return estimator