!5307 change enable_data_sink value to true for transformer

Merge pull request !5307 from yuchaojie/transformer2

model_zoo/official/nlp/transformer/scripts/run_distribute_train_ascend.sh

@@ -52,7 +52,7 @@ do
     --enable_save_ckpt="true" \
     --enable_lossscale="true" \
     --do_shuffle="true" \
-    --enable_data_sink="false" \
+    --enable_data_sink="true" \
     --checkpoint_path="" \
     --save_checkpoint_steps=2500 \
     --save_checkpoint_num=30 \

model_zoo/official/nlp/transformer/scripts/run_standalone_train_ascend.sh

@@ -37,7 +37,7 @@ python train.py  \
     --enable_save_ckpt="true" \
     --enable_lossscale="true" \
     --do_shuffle="true" \
-    --enable_data_sink="false" \
+    --enable_data_sink="true" \
     --checkpoint_path="" \
     --save_checkpoint_steps=2500 \
     --save_checkpoint_num=30 \

model_zoo/official/nlp/transformer/train.py

@@ -170,7 +170,8 @@ def run_transformer_train():
 
     netwithgrads.set_train(True)
     model = Model(netwithgrads)
-    model.train(args.epoch_size, dataset, callbacks=callbacks, dataset_sink_mode=(args.enable_data_sink == "true"))
+    model.train(args.epoch_size, dataset, callbacks=callbacks, dataset_sink_mode=(args.enable_data_sink == "true"),
+                sink_size=args.save_checkpoint_steps)
 
 if __name__ == '__main__':
     run_transformer_train()