From a2181cab18a5fd865c4bae455078cc53af4761db Mon Sep 17 00:00:00 2001
From: liu zhengxi <380185688@qq.com>
Date: Fri, 18 Dec 2020 19:08:26 +0800
Subject: [PATCH] work around hung (#5119)

---
 .../transformer/configs/transformer.base.yaml                   | 2 +-
 .../transformer/configs/transformer.big.yaml                    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/PaddleNLP/examples/machine_translation/transformer/configs/transformer.base.yaml b/PaddleNLP/examples/machine_translation/transformer/configs/transformer.base.yaml
index 30dff6dc..7ea9ebbe 100644
--- a/PaddleNLP/examples/machine_translation/transformer/configs/transformer.base.yaml
+++ b/PaddleNLP/examples/machine_translation/transformer/configs/transformer.base.yaml
@@ -24,7 +24,7 @@ use_gpu: True
 
 # Args for reader, see reader.py for details
 pool_size: 200000
-sort_type: "pool"
+sort_type: "global"
 batch_size: 4096
 infer_batch_size: 8
 
diff --git a/PaddleNLP/examples/machine_translation/transformer/configs/transformer.big.yaml b/PaddleNLP/examples/machine_translation/transformer/configs/transformer.big.yaml
index 5315884e..d458f4c7 100644
--- a/PaddleNLP/examples/machine_translation/transformer/configs/transformer.big.yaml
+++ b/PaddleNLP/examples/machine_translation/transformer/configs/transformer.big.yaml
@@ -24,7 +24,7 @@ use_gpu: True
 
 # Args for reader, see reader.py for details
 pool_size: 200000
-sort_type: "pool"
+sort_type: "global"
 batch_size: 4096
 infer_batch_size: 8
 
-- 
GitLab