Add memory limit (#552)

This aids GKE autoprovisioning in creation of appropriate node sizes
2019-05-14 17:54:18 -07:00 · 2019-05-14 17:54:18 -07:00 · 1ebcd33881
parent ed38a1a35c
commit 1ebcd33881
2 changed files with 1 additions and 0 deletions
--- a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py
+++ b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py
@ -55,6 +55,7 @@ def gh_summ(  #pylint: disable=unused-argument
      )
  serve.after(train)
  train.set_gpu_limit(4)
+  train.set_memory_limit('48G')

  with dsl.Condition(train.output == 'true'):
    webapp = dsl.ContainerOp(
--- a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz
+++ b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz