From 9ece54aecfb011371df6c1fcf3d0b66969ffa7c4 Mon Sep 17 00:00:00 2001 From: Marcio Vinicius dos Santos Date: Fri, 13 Apr 2018 07:56:35 -0700 Subject: [PATCH] Update test_tf_cifar.py --- tests/integ/test_tf_cifar.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integ/test_tf_cifar.py b/tests/integ/test_tf_cifar.py index 648d6133d3..8f270026db 100644 --- a/tests/integ/test_tf_cifar.py +++ b/tests/integ/test_tf_cifar.py @@ -37,7 +37,7 @@ def test_cifar(sagemaker_session, tf_full_version): dataset_path = os.path.join(DATA_DIR, 'cifar_10', 'data') estimator = TensorFlow(entry_point='resnet_cifar_10.py', source_dir=script_path, role='SageMakerRole', - framework_version=tf_full_version, training_steps=20, evaluation_steps=5, + framework_version=tf_full_version, training_steps=10, evaluation_steps=5, train_instance_count=2, train_instance_type='ml.p2.xlarge', sagemaker_session=sagemaker_session, base_job_name='test-cifar')