change enable_data_sink value to true for transformer

pull/5307/head
yuchaojie 5 years ago
parent 5fc18b9833
commit 15b366307e

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

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

@ -166,7 +166,8 @@ def run_transformer_train():
netwithgrads.set_train(True) netwithgrads.set_train(True)
model = Model(netwithgrads) 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__': if __name__ == '__main__':
run_transformer_train() run_transformer_train()

Loading…
Cancel
Save