update transformer model train

update ParallelExecutor params
Superjomn · May 4, 2018 · 548c67e · 548c67e
1 parent 8b2a70e
commit 548c67e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformer/train.py b/transformer/train.py
@@ -215,7 +215,7 @@ def test(exe):
     train_exe = fluid.ParallelExecutor(
         use_cuda=TrainTaskConfig.use_gpu,
         loss_name=sum_cost.name,
-        customize_loss_grad=True)
+        use_default_grad_scale=False)
 
     test_exe = fluid.ParallelExecutor(
         use_cuda=TrainTaskConfig.use_gpu,