diff --git a/dags/solutions_team/solutionsteam_tf_nightly_supported.py b/dags/solutions_team/solutionsteam_tf_nightly_supported.py index 5dc2bd6a..13abe496 100644 --- a/dags/solutions_team/solutionsteam_tf_nightly_supported.py +++ b/dags/solutions_team/solutionsteam_tf_nightly_supported.py @@ -136,7 +136,7 @@ tpu_version=TpuVersion.V5P, tpu_cores=32, tpu_zone=Zone.US_EAST5_A.value, - time_out_in_min=60, + time_out_in_min=120, bottom_mlp=[512, 256, embedding_dim], embedding_dim=embedding_dim, train_steps=10000, diff --git a/dags/solutions_team/solutionsteam_tf_se_nightly_supported.py b/dags/solutions_team/solutionsteam_tf_se_nightly_supported.py index 663d96d7..ecb51554 100644 --- a/dags/solutions_team/solutionsteam_tf_se_nightly_supported.py +++ b/dags/solutions_team/solutionsteam_tf_se_nightly_supported.py @@ -97,7 +97,7 @@ tpu_version=TpuVersion.V2, tpu_cores=8, tpu_zone=Zone.US_CENTRAL1_C.value, - time_out_in_min=60, + time_out_in_min=120, bottom_mlp=[512, 256, embedding_dim], embedding_dim=embedding_dim, train_steps=10000, @@ -111,7 +111,7 @@ tpu_version=TpuVersion.V2, tpu_cores=32, tpu_zone=Zone.US_CENTRAL1_A.value, - time_out_in_min=60, + time_out_in_min=120, bottom_mlp=[512, 256, embedding_dim], embedding_dim=embedding_dim, train_steps=256054, @@ -126,7 +126,7 @@ tpu_version=TpuVersion.V4, tpu_cores=8, tpu_zone=Zone.US_CENTRAL2_B.value, - time_out_in_min=60, + time_out_in_min=120, bottom_mlp=[512, 256, embedding_dim], embedding_dim=embedding_dim, train_steps=10000, @@ -140,7 +140,7 @@ tpu_version=TpuVersion.V4, tpu_cores=32, tpu_zone=Zone.US_CENTRAL2_B.value, - time_out_in_min=60, + time_out_in_min=120, bottom_mlp=[512, 256, embedding_dim], embedding_dim=embedding_dim, train_steps=128000,