From 8217a8c253a4095ff6ff2d9ceb8905432cdf1f90 Mon Sep 17 00:00:00 2001 From: Srihari Date: Tue, 6 Aug 2024 11:01:08 +0530 Subject: [PATCH] Update HPO Ray Tune GPU test --- tests/odh/resources/mnist_hpo_raytune.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/odh/resources/mnist_hpo_raytune.ipynb b/tests/odh/resources/mnist_hpo_raytune.ipynb index 5ad27b9c..ed00c762 100644 --- a/tests/odh/resources/mnist_hpo_raytune.ipynb +++ b/tests/odh/resources/mnist_hpo_raytune.ipynb @@ -170,7 +170,7 @@ " \"working_dir\": \"/opt/app-root/notebooks/..data\",\n", " \"pip\": \"/opt/app-root/notebooks/hpo_raytune_requirements.txt\",\n", " },\n", - " entrypoint_num_gpus=num_gpus\n", + " # entrypoint_num_gpus is not required here as the mnist_hpo script executes in parallel and requires more GPUs for each iteration\n", ")" ] },