Update solutionsteam_tf_nightly_supported.py (#498)

GoogleCloudPlatform · Dec 5, 2024 · 15da729 · 15da729
1 parent c05df73
commit 15da729
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/solutions_team/solutionsteam_tf_nightly_supported.py b/dags/solutions_team/solutionsteam_tf_nightly_supported.py
@@ -131,7 +131,7 @@
       network=V5_NETWORKS,
       subnetwork=V5P_SUBNETWORKS,
       runtime_version=RuntimeVersion.V2_ALPHA_TPUV5.value,
-      global_batch_size=8196,
+      global_batch_size=8192,
   )
 
   embedding_dim = 128