Increase timeouts for tensorflow tests (#421)

Co-authored-by: Allen Wang <[email protected]>
GoogleCloudPlatform · Oct 3, 2024 · d1148d8 · d1148d8
1 parent 999f68a
commit d1148d8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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,