Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
data/README.md
View file @ 0a141e3f
...@@ -18,9 +18,9 @@ $ python preprocess.py --source-lang de --target-lang en \ ...@@ -18,9 +18,9 @@ $ python preprocess.py --source-lang de --target-lang en \
--trainpref $TEXT/train --validpref $TEXT/valid --testpref $TEXT/test \ --trainpref $TEXT/train --validpref $TEXT/valid --testpref $TEXT/test \
--destdir data-bin/iwslt14.tokenized.de-en --destdir data-bin/iwslt14.tokenized.de-en
# Train the model: # Train the model (better for a single GPU setup):
$ mkdir -p checkpoints/fconv $ 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 \ --lr 0.25 --clip-norm 0.1 --dropout 0.2 --max-tokens 4000 \
--criterion label_smoothed_cross_entropy --label-smoothing 0.1 \ --criterion label_smoothed_cross_entropy --label-smoothing 0.1 \
--lr-scheduler fixed --force-anneal 200 \ --lr-scheduler fixed --force-anneal 200 \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment