From ee54640f235fc87974afb3f23c9ed451d72ed323 Mon Sep 17 00:00:00 2001 From: xuezhong Date: Fri, 2 Nov 2018 07:17:25 +0000 Subject: [PATCH] default setting --- fluid/PaddleNLP/machine_reading_comprehension/.run_ce.sh | 2 +- fluid/PaddleNLP/machine_reading_comprehension/_ce.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/fluid/PaddleNLP/machine_reading_comprehension/.run_ce.sh b/fluid/PaddleNLP/machine_reading_comprehension/.run_ce.sh index 715d5bb3..4340aa73 100644 --- a/fluid/PaddleNLP/machine_reading_comprehension/.run_ce.sh +++ b/fluid/PaddleNLP/machine_reading_comprehension/.run_ce.sh @@ -39,7 +39,7 @@ export CUDA_VISIBLE_DEVICES=$cudaid train | python _ce.py -cudaid=${transformer_m:=0,1} # use 0,1 card as default +cudaid=${transformer_m:=0,1,2,3} # use 0,1,2,3 card as default export CUDA_VISIBLE_DEVICES=$cudaid train | python _ce.py diff --git a/fluid/PaddleNLP/machine_reading_comprehension/_ce.py b/fluid/PaddleNLP/machine_reading_comprehension/_ce.py index 24243846..cff13c87 100644 --- a/fluid/PaddleNLP/machine_reading_comprehension/_ce.py +++ b/fluid/PaddleNLP/machine_reading_comprehension/_ce.py @@ -11,10 +11,10 @@ train_cost_card1_kpi = CostKpi('train_cost_card1', 0.02, 0, actived=True) test_cost_card1_kpi = CostKpi('test_cost_card1', 0.005, 0, actived=True) train_duration_card1_kpi = DurationKpi( 'train_duration_card1', 0.06, 0, actived=True) -train_cost_card4_kpi = CostKpi('train_cost_card2', 0.01, 0, actived=True) -test_cost_card4_kpi = CostKpi('test_cost_card2', 0.005, 0, actived=True) +train_cost_card4_kpi = CostKpi('train_cost_card4', 0.01, 0, actived=True) +test_cost_card4_kpi = CostKpi('test_cost_card4', 0.005, 0, actived=True) train_duration_card4_kpi = DurationKpi( - 'train_duration_card2', 0.06, 0, actived=True) + 'train_duration_card4', 0.06, 0, actived=True) tracking_kpis = [ train_cost_card1_kpi, -- GitLab