Update HPO Ray Tune GPU test

opendatahub-io · Aug 7, 2024 · 8217a8c · 8217a8c
1 parent 5c39a1b
commit 8217a8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",
     ")"
    ]
   },