Skip to content

Commit a13520c

Browse files
committed
Update training and evaluation step
1 parent a31074e commit a13520c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_tf_keras.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def test_keras(sagemaker_session, tf_full_version):
3131
source_dir=script_path,
3232
role='SageMakerRole',
3333
hyperparameters={'learning_rate': 1e-4, 'decay': 1e-6},
34-
training_steps=1000, evaluation_steps=100,
34+
training_steps=500, evaluation_steps=5,
3535
train_instance_count=1, train_instance_type='ml.c4.xlarge')
3636

3737
inputs = estimator.sagemaker_session.upload_data(path=dataset_path, key_prefix='data/cifar10')

0 commit comments

Comments
 (0)