From c4cee259596a7e50e528d7375a08c4cafc94e1f6 Mon Sep 17 00:00:00 2001 From: malin10 Date: Wed, 3 Jun 2020 13:48:48 +0800 Subject: [PATCH] bug fix --- core/trainers/single_trainer.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/core/trainers/single_trainer.py b/core/trainers/single_trainer.py index abff7492..305707d6 100755 --- a/core/trainers/single_trainer.py +++ b/core/trainers/single_trainer.py @@ -248,10 +248,9 @@ class SingleTrainer(TranspileTrainer): _exe_strategy = fluid.ExecutionStrategy() # 0: kCoeffNumDevice; 1: One; 2: Customized + gradient_scale_strategy = fluid.BuildStrategy.GradientScaleStrategy.CoeffNumDevice _gradient_scale_strategy = model_dict.get("gradient_scale_strategy", 0) - if _gradient_scale_strategy == 0: - gradient_scale_strategy = fluid.BuildStrategy.GradientScaleStrategy.CoeffNumDevice - elif _gradient_scale_strategy == 1: + if _gradient_scale_strategy == 1: gradient_scale_strategy = fluid.BuildStrategy.GradientScaleStrategy.One elif _gradient_scale_strategy == 2: gradient_scale_strategy = fluid.BuildStrategy.GradientScaleStrategy.Customized -- GitLab