diff --git a/fluid/PaddleNLP/machine_reading_comprehension/.run_ce.sh b/fluid/PaddleNLP/machine_reading_comprehension/.run_ce.sh index 715d5bb378ce8989fe9dbda04722c86da42cd3fa..4340aa735302e343973b3084af9f4edd2b3c75d9 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 24243846ee1f321cb449573f1a3729346edd1035..cff13c8722007987a3cd82f1298206248963e45a 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,