Add resnet50 + convergence + v4_8 test

GoogleCloudPlatform · Nov 13, 2023 · 2967337 · 2967337
1 parent 49e8b9e
commit 2967337
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/pytorch/nightly/resnet50-mp.libsonnet b/tests/pytorch/nightly/resnet50-mp.libsonnet
@@ -209,5 +209,6 @@ local tpus = import 'templates/tpus.libsonnet';
     // SPMD
     resnet50 + functional + v4_8 + timeouts.Hours(2) + spmd(['batch']),
     resnet50 + functional + v4_8 + timeouts.Hours(2) + spmd(['spatial']),
+    resnet50 + convergence + v4_8 + timeouts.Hours(14) + spmd(['batch']),
   ],
 }