From ad55b0a246b01586e5e8f110b7d2dc5e5a5c3c1a Mon Sep 17 00:00:00 2001 From: Amy Date: Thu, 3 Oct 2019 09:12:09 -0700 Subject: [PATCH] change gpu limit (#651) --- .../pipelines/example_pipelines/gh_summ.py | 2 +- .../example_pipelines/gh_summ.py.tar.gz | Bin 1700 -> 1700 bytes 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py index 0398ae2c..bf0bde1a 100644 --- 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 index f9a7a864aa7270a372b6f5853d14bc34eb75ac83..b285e71f412e579056a99d16c0c05b517d72446b 100644 GIT binary patch delta 135 zcmV;20C@kT4WtbRABzYGSY4H|2Mz{Z9i#nrKIN=Cme)UfKW$r6^(x}+uUYGBx#3ew zC9%?wEQrj7`m^nx)Zd@2Z{*eAm1=W2fhO=_9CCn@C8c-%X+`PR8*d+MNvBGG!c(v7 pB&{n;L2{M+Mo85>(361%7n6?$36qZo1`Nx~%76J&qJRKU001DIKlK0r delta 133 zcmV;00DAwV4WtbRABzYGl&piX2Mz{Y{O;TNl(X(wUjOX