From 53f69cf8558779d9e520f36833e19d64226edfcd Mon Sep 17 00:00:00 2001
From: Winston Dong <win@amazon.com>
Date: Sat, 5 May 2018 00:33:34 +0000
Subject: [PATCH] Increase train steps for cifar distributed test to mitigate
 race condition

---
 test/functional/test_distributed.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/functional/test_distributed.py b/test/functional/test_distributed.py
index 27a1049b..16fb83e1 100644
--- a/test/functional/test_distributed.py
+++ b/test/functional/test_distributed.py
@@ -50,7 +50,7 @@ def test_distributed(instance_type, sagemaker_session, docker_image_uri):
         estimator = MyEstimator(entry_point='resnet_cifar_10.py',
                                 source_dir=script_path,
                                 role='SageMakerRole',
-                                training_steps=2,
+                                training_steps=500,
                                 evaluation_steps=1,
                                 train_instance_count=2,
                                 train_instance_type=instance_type,