fix a typo and remove a too restrictive limit (kubeflow#1382)

StatCan · Aug 17, 2018 · cf4e136 · cf4e136
1 parent 398e38d
commit cf4e136
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/kubeflow/examples/prototypes/tf-batch-predict.jsonnet b/kubeflow/examples/prototypes/tf-batch-predict.jsonnet
@@ -87,9 +87,7 @@ local tfBatchPredictBase = {
 
                 resources: {
                   limits: {
-                    memory: "4Gi",
-                    cpu: "4",
-                    [if base.params.numGpus > 0 then "nvidia/com/gpu"]: base.params.numGpus,
+                    [if base.params.numGpus > 0 then "nvidia.com/gpu"]: base.params.numGpus,
                   },
                 },