Skip to content

Commit cca140d

Browse files
winstonawslaurenyu
authored andcommitted
Increase train steps for cifar distributed test to mitigate race condition (#37)
1 parent c88b1ce commit cca140d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/functional/test_distributed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def test_distributed(instance_type, sagemaker_session, docker_image_uri):
5050
estimator = MyEstimator(entry_point='resnet_cifar_10.py',
5151
source_dir=script_path,
5252
role='SageMakerRole',
53-
training_steps=2,
53+
training_steps=500,
5454
evaluation_steps=1,
5555
train_instance_count=2,
5656
train_instance_type=instance_type,

0 commit comments

Comments
 (0)