change gpu limit (#651)

2019-10-03 09:12:09 -07:00 · 2019-10-03 09:12:09 -07:00 · ad55b0a246
parent 351aaf81c7
commit ad55b0a246
2 changed files with 1 additions and 1 deletions
--- 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'):
--- a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz
+++ b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz