diff --git a/dygraph/lac/run.sh b/dygraph/lac/run.sh index d8ca0f5d18a78fb92b780fff22e34421f718fb74..f994764d6e75c0d1ee9f60656ec4b8b4c26b7e2a 100755 --- a/dygraph/lac/run.sh +++ b/dygraph/lac/run.sh @@ -9,9 +9,9 @@ python -m paddle.distributed.launch --selected_gpus=0,1,2,3 train.py \ --test_data ./data/test.tsv \ --model_save_dir ./padding_models \ --validation_steps 1000 \ - --save_steps 10000 \ - --print_steps 200 \ - --batch_size 400 \ + --save_steps 1 \ + --print_steps 1 \ + --batch_size 32 \ --epoch 10 \ --traindata_shuffle_buffer 20000 \ --word_emb_dim 128 \