removing extra gpu check as discussed with johnu

deepanker13 · Jan 10, 2024 · 4dd21c7 · 4dd21c7
1 parent b2dbcd0
commit 4dd21c7
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/sdk/python/kubeflow/training/api/training_client.py b/sdk/python/kubeflow/training/api/training_client.py
@@ -139,11 +139,8 @@ def train(
 
         if isinstance(resources_per_worker, dict):
             if "gpu" in resources_per_worker:
-                if (
-                    resources_per_worker["gpu"] is not None
-                    and (num_procs_per_worker > resources_per_worker["gpu"])
-                ) or (
-                    resources_per_worker["gpu"] is None and num_procs_per_worker != 0
+                if resources_per_worker["gpu"] is not None and (
+                    num_procs_per_worker > resources_per_worker["gpu"]
                 ):
                     raise ValueError(
                         "Insufficient gpu resources allocated to the container."