File tree Expand file tree Collapse file tree 4 files changed +13
-12
lines changed Expand file tree Collapse file tree 4 files changed +13
-12
lines changed Original file line number Diff line number Diff line change 1
1
#! /usr/bin/env bash
2
2
set -ex
3
3
THIS_DIR=$( dirname $( readlink -f $0 ) )
4
- cd $THIS_DIR /../..
4
+ cd $THIS_DIR /../../..
5
5
6
6
if [ ! -d " wmt14_en_de" ]; then
7
7
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
10
10
fi
11
11
12
12
deepspeed ${THIS_DIR} /ds_fairseq.py \
13
- . /wmt14_en_de/ \
13
+ /tmp /wmt14_en_de/ \
14
14
--user-dir ${THIS_DIR} /../../../lightseq/training/fs_modules \
15
15
--arch ls_transformer_wmt_en_de_big_t2t --share-decoder-input-output-embed \
16
16
--optimizer ls_adam --adam-betas ' (0.9, 0.98)' --clip-norm 0.0 \
Original file line number Diff line number Diff line change 1
1
#! /usr/bin/env bash
2
2
set -ex
3
3
THIS_DIR=$( dirname $( readlink -f $0 ) )
4
- cd $THIS_DIR /../../
4
+ cd $THIS_DIR /../../..
5
5
6
6
if [ ! -d " wmt14_en_de" ]; then
7
7
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
10
10
fi
11
11
12
- lightseq-train . /wmt14_en_de/ \
12
+ lightseq-train /tmp /wmt14_en_de/ \
13
13
--task translation \
14
14
--arch ls_transformer_wmt_en_de_big_t2t --share-decoder-input-output-embed \
15
15
--optimizer ls_adam --adam-betas ' (0.9, 0.98)' \
Original file line number Diff line number Diff line change 1
1
#! /usr/bin/env bash
2
2
set -ex
3
3
THIS_DIR=$( dirname $( readlink -f $0 ) )
4
- cd $THIS_DIR /../../
4
+ cd $THIS_DIR /../../..
5
5
6
6
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
9
10
fi
10
11
11
- fairseq-train . /wmt14_en_de/ \
12
+ fairseq-train /tmp /wmt14_en_de/ \
12
13
--arch transformer_wmt_en_de_big_t2t --share-decoder-input-output-embed \
13
14
--optimizer adam --adam-betas ' (0.9, 0.98)' --clip-norm 0.0 \
14
15
--lr 5e-4 --lr-scheduler inverse_sqrt --warmup-updates 4000 \
You can’t perform that action at this time.
0 commit comments