Unverified Commit 0a141e3f authored by Runqi Yang's avatar Runqi Yang Committed by GitHub
Browse files

Update training command for IWSLT14 

specify a single GPU setup for IWSLT14
parent 435ed351

data/README.md

+2 −2
Changes for data/README.md: 2 added lines, 2 removed lines.
Original line number Diff line number Diff line
@@ -18,9 +18,9 @@ $ python preprocess.py --source-lang de --target-lang en \ 
  --trainpref $TEXT/train --validpref $TEXT/valid --testpref $TEXT/test \

  --destdir data-bin/iwslt14.tokenized.de-en



# Train the model:

# Train the model (better for a single GPU setup):

$ mkdir -p checkpoints/fconv

$ python train.py data-bin/iwslt14.tokenized.de-en \

$ CUDA_VISIBLE_DEVICES=0 python train.py data-bin/iwslt14.tokenized.de-en \

  --lr 0.25 --clip-norm 0.1 --dropout 0.2 --max-tokens 4000 \

  --criterion label_smoothed_cross_entropy --label-smoothing 0.1 \

  --lr-scheduler fixed --force-anneal 200 \