@@ -6,6 +6,14 @@ if [ ${MODE} = "lite_train_infer" ]; then
66 # Hence, set maximum output length is 16.
77 sed -i " s/^max_out_len.*/max_out_len: 16/g" configs/transformer.base.yaml
88 sed -i " s/^max_out_len.*/max_out_len: 16/g" configs/transformer.big.yaml
9+
10+ sed -i " s/^random_seed:.*/random_seed: 128/g" configs/transformer.base.yaml
11+ sed -i " s/^shuffle_batch:.*/shuffle_batch: False/g" configs/transformer.base.yaml
12+ sed -i " s/^shuffle:.*/shuffle: False/g" configs/transformer.base.yaml
13+
14+ sed -i " s/^random_seed:.*/random_seed: 128/g" configs/transformer.big.yaml
15+ sed -i " s/^shuffle_batch:.*/shuffle_batch: False/g" configs/transformer.big.yaml
16+ sed -i " s/^shuffle:.*/shuffle: False/g" configs/transformer.big.yaml
917 # Data set prepared.
1018 if [ ! -f WMT14.en-de.partial.tar.gz ]; then
1119 wget https://paddlenlp.bj.bcebos.com/datasets/WMT14.en-de.partial.tar.gz
@@ -51,6 +59,14 @@ elif [ ${MODE} = "whole_infer" ]; then
5159 sed -i " s/^max_out_len.*/max_out_len: 256/g" configs/transformer.base.yaml
5260 sed -i " s/^max_out_len.*/max_out_len: 1024/g" configs/transformer.big.yaml
5361
62+ sed -i " s/^random_seed:.*/random_seed: None/g" configs/transformer.base.yaml
63+ sed -i " s/^shuffle_batch:.*/shuffle_batch: True/g" configs/transformer.base.yaml
64+ sed -i " s/^shuffle:.*/shuffle: True/g" configs/transformer.base.yaml
65+
66+ sed -i " s/^random_seed:.*/random_seed: None/g" configs/transformer.big.yaml
67+ sed -i " s/^shuffle_batch:.*/shuffle_batch: True/g" configs/transformer.big.yaml
68+ sed -i " s/^shuffle:.*/shuffle: True/g" configs/transformer.big.yaml
69+
5470 # Trained transformer base model checkpoint.
5571 # For infer.
5672 if [ ! -f tranformer-base-wmt_ende_bpe.tar.gz ]; then
@@ -108,6 +124,14 @@ elif [ ${MODE} = "whole_train_infer" ]; then
108124 sed -i " s/^max_out_len.*/max_out_len: 256/g" configs/transformer.base.yaml
109125 sed -i " s/^max_out_len.*/max_out_len: 1024/g" configs/transformer.big.yaml
110126
127+ sed -i " s/^random_seed:.*/random_seed: None/g" configs/transformer.base.yaml
128+ sed -i " s/^shuffle_batch:.*/shuffle_batch: True/g" configs/transformer.base.yaml
129+ sed -i " s/^shuffle:.*/shuffle: True/g" configs/transformer.base.yaml
130+
131+ sed -i " s/^random_seed:.*/random_seed: None/g" configs/transformer.big.yaml
132+ sed -i " s/^shuffle_batch:.*/shuffle_batch: True/g" configs/transformer.big.yaml
133+ sed -i " s/^shuffle:.*/shuffle: True/g" configs/transformer.big.yaml
134+
111135 # Whole data set prepared.
112136 if [ ! -f WMT14.en-de.tar.gz ]; then
113137 wget https://paddlenlp.bj.bcebos.com/datasets/WMT14.en-de.tar.gz
@@ -153,6 +177,14 @@ else # infer
153177 sed -i " s/^max_out_len.*/max_out_len: 256/g" configs/transformer.base.yaml
154178 sed -i " s/^max_out_len.*/max_out_len: 1024/g" configs/transformer.big.yaml
155179
180+ sed -i " s/^random_seed:.*/random_seed: None/g" configs/transformer.base.yaml
181+ sed -i " s/^shuffle_batch:.*/shuffle_batch: True/g" configs/transformer.base.yaml
182+ sed -i " s/^shuffle:.*/shuffle: True/g" configs/transformer.base.yaml
183+
184+ sed -i " s/^random_seed:.*/random_seed: None/g" configs/transformer.big.yaml
185+ sed -i " s/^shuffle_batch:.*/shuffle_batch: True/g" configs/transformer.big.yaml
186+ sed -i " s/^shuffle:.*/shuffle: True/g" configs/transformer.big.yaml
187+
156188 # Trained transformer base model checkpoint.
157189 if [ ! -f tranformer-base-wmt_ende_bpe.tar.gz ]; then
158190 wget https://paddlenlp.bj.bcebos.com/models/transformers/transformer/tranformer-base-wmt_ende_bpe.tar.gz
0 commit comments