change gpu limit (kubeflow#651)

AmandeepSinghCS · Oct 3, 2019 · ad55b0a · ad55b0a
1 parent 351aaf8
commit ad55b0a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py
@@ -95,7 +95,7 @@ def gh_summ(  #pylint: disable=unused-argument
   train.after(copydata)
   log_model.after(train)
   serve.after(train)
-  train.set_gpu_limit(4)
+  train.set_gpu_limit(1)
   train.set_memory_limit('48G')
 
   with dsl.Condition(train.output == 'true'):

diff --git a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz