diff --git a/PaddleNLP/examples/machine_translation/transformer/configs/transformer.base.yaml b/PaddleNLP/examples/machine_translation/transformer/configs/transformer.base.yaml index 30dff6dca05d0454ff53096dbeac691a0eb1f7d6..7ea9ebbe7141963a2934825e3c9200a1ec40d3f2 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 5315884e3172db3d12dff84885784c32723faf63..d458f4c7eb15124c3bb82fb8858528563a32eba2 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