Merge pull request #6 from Superjomn/kolinwei-patch-1

update transformer model train
Superjomn · May 4, 2018 · 2aca4bd · 2aca4bd
2 parents 8b2a70e + 548c67e
commit 2aca4bd
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,