Skip to content

Commit fc901a6

Browse files
committed
Update train_max_run
1 parent b5d2d13 commit fc901a6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/integ/test_tf_keras.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@ def test_keras(sagemaker_session, tf_full_version):
3333
role='SageMakerRole',
3434
hyperparameters={'learning_rate': 1e-4, 'decay': 1e-6},
3535
training_steps=500, evaluation_steps=5,
36-
train_instance_count=1, train_instance_type='ml.c4.xlarge')
36+
train_instance_count=1, train_instance_type='ml.c4.xlarge',
37+
train_max_run=45 * 60)
3738

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

0 commit comments

Comments
 (0)