Skip to content

Commit

Permalink
stories data preprocessing needs padding factor 1 to match pretrained…
Browse files Browse the repository at this point in the history
… model, updating readme
  • Loading branch information
huihuifan committed Jul 20, 2018
1 parent eaa576b commit 28adb20
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/stories/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ Example usage:
$ TEXT=examples/stories/writingPrompts
$ python preprocess.py --source-lang wp_source --target-lang wp_target \
--trainpref $TEXT/train --validpref $TEXT/valid --testpref $TEXT/test \
--destdir data-bin/writingPrompts --thresholdtgt 10 --thresholdsrc 10
--destdir data-bin/writingPrompts --padding-factor 1 --thresholdtgt 10 --thresholdsrc 10
# Train the model:
$ python train.py data-bin/writingPrompts -a fconv_self_att_wp --lr 0.25 --clip-norm 0.1 --max-tokens 1500 --lr-scheduler reduce_lr_on_plateau --decoder-attention True --encoder-attention False --criterion label_smoothed_cross_entropy --weight-decay .0000001 --label-smoothing 0 --source-lang wp_source --target-lang wp_target --gated-attention True --self-attention True --project-input True --pretrained False
Expand Down

0 comments on commit 28adb20

Please sign in to comment.