Skip to content

Commit 9c05fc9

Browse files
committed
modify running script
1 parent 6186044 commit 9c05fc9

File tree

4 files changed

+13
-12
lines changed

4 files changed

+13
-12
lines changed

examples/__init__.py

Whitespace-only changes.

examples/training/deepspeed/ds_fairseq_wmt14en2de.sh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
#!/usr/bin/env bash
22
set -ex
33
THIS_DIR=$(dirname $(readlink -f $0))
4-
cd $THIS_DIR/../..
4+
cd $THIS_DIR/../../..
55

66
if [ ! -d "wmt14_en_de" ]; then
77
echo "Downloading dataset"
8-
wget http://sf3-ttcdn-tos.pstatp.com/obj/nlp-opensource/lightseq/wmt_data/databin_wmt14_en_de.tar.gz
9-
tar -zxvf databin_wmt14_en_de.tar.gz && rm databin_wmt14_en_de.tar.gz
8+
wget http://sf3-ttcdn-tos.pstatp.com/obj/nlp-opensource/lightseq/wmt_data/databin_wmt14_en_de.tar.gz -P /tmp
9+
tar -zxvf /tmp/databin_wmt14_en_de.tar.gz && rm /tmp/databin_wmt14_en_de.tar.gz
1010
fi
1111

1212
deepspeed ${THIS_DIR}/ds_fairseq.py \
13-
./wmt14_en_de/ \
13+
/tmp/wmt14_en_de/ \
1414
--user-dir ${THIS_DIR}/../../../lightseq/training/fs_modules \
1515
--arch ls_transformer_wmt_en_de_big_t2t --share-decoder-input-output-embed \
1616
--optimizer ls_adam --adam-betas '(0.9, 0.98)' --clip-norm 0.0 \

examples/training/fairseq/ls_fairseq_wmt14en2de.sh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
11
#!/usr/bin/env bash
22
set -ex
33
THIS_DIR=$(dirname $(readlink -f $0))
4-
cd $THIS_DIR/../../
4+
cd $THIS_DIR/../../..
55

66
if [ ! -d "wmt14_en_de" ]; then
77
echo "Downloading dataset"
8-
wget http://sf3-ttcdn-tos.pstatp.com/obj/nlp-opensource/lightseq/wmt_data/databin_wmt14_en_de.tar.gz
9-
tar -zxvf databin_wmt14_en_de.tar.gz && rm databin_wmt14_en_de.tar.gz
8+
wget http://sf3-ttcdn-tos.pstatp.com/obj/nlp-opensource/lightseq/wmt_data/databin_wmt14_en_de.tar.gz -P /tmp
9+
tar -zxvf /tmp/databin_wmt14_en_de.tar.gz && rm /tmp/databin_wmt14_en_de.tar.gz
1010
fi
1111

12-
lightseq-train ./wmt14_en_de/ \
12+
lightseq-train /tmp/wmt14_en_de/ \
1313
--task translation \
1414
--arch ls_transformer_wmt_en_de_big_t2t --share-decoder-input-output-embed \
1515
--optimizer ls_adam --adam-betas '(0.9, 0.98)' \

examples/training/fairseq/native_fairseq_wmt14en2de.sh

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,15 @@
11
#!/usr/bin/env bash
22
set -ex
33
THIS_DIR=$(dirname $(readlink -f $0))
4-
cd $THIS_DIR/../../
4+
cd $THIS_DIR/../../..
55

66
if [ ! -d "wmt14_en_de" ]; then
7-
wget http://sf3-ttcdn-tos.pstatp.com/obj/nlp-opensource/lightseq/wmt_data/databin_wmt14_en_de.tar.gz
8-
tar -zxvf databin_wmt14_en_de.tar.gz && rm databin_wmt14_en_de.tar.gz
7+
echo "Downloading dataset"
8+
wget http://sf3-ttcdn-tos.pstatp.com/obj/nlp-opensource/lightseq/wmt_data/databin_wmt14_en_de.tar.gz -P /tmp
9+
tar -zxvf /tmp/databin_wmt14_en_de.tar.gz && rm /tmp/databin_wmt14_en_de.tar.gz
910
fi
1011

11-
fairseq-train ./wmt14_en_de/ \
12+
fairseq-train /tmp/wmt14_en_de/ \
1213
--arch transformer_wmt_en_de_big_t2t --share-decoder-input-output-embed \
1314
--optimizer adam --adam-betas '(0.9, 0.98)' --clip-norm 0.0 \
1415
--lr 5e-4 --lr-scheduler inverse_sqrt --warmup-updates 4000 \

0 commit comments

Comments
 (0)