File size: 4,734 Bytes
632f87a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 |
save_data: run
## Where the vocab(s) will be written
src_vocab: run/bpe.vocab.src
tgt_vocab: run/bpe.vocab.tgt
overwrite: True
# Corpus opts:
data:
europarl:
path_src: ../DGTcorpora_tokenized/en_gl/europarl/partitions/en_train.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/europarl/partitions/gl_train.txt
transforms: [bpe, filtertoolong]
weight: 120
opensub:
path_src: ../DGTcorpora_tokenized/en_gl/opensub/partitions/en_train.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/opensub/partitions/gl_train.txt
transforms: [bpe, filtertoolong]
weight: 152
opus:
path_src: ../DGTcorpora_tokenized/en_gl/opus/partitions/en_train.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/opus/partitions/gl_train.txt
transforms: [bpe, filtertoolong]
weight: 160
ted2020:
path_src: ../DGTcorpora_tokenized/en_gl/ted2020/partitions/en_train.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/ted2020/partitions/gl_train.txt
transforms: [bpe, filtertoolong]
weight: 10
corgaback:
path_src: ../DGTcorpora_tokenized/en_gl/corgaback/partitions/en_train.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/corgaback/partitions/gl_train.txt
transforms: [bpe, filtertoolong]
weight: 15
ccmatrix:
path_src: ../DGTcorpora_tokenized/en_gl/ccmatrix/en_tok_dbo.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/ccmatrix/gl_tok_dbo.txt
transforms: [bpe, filtertoolong]
weight: 380 ##75 ## 25000000/13000000 = 2; 760/2 = 380 * 5 = 1900 (380/5=75)
wikimatrix:
path_src: ../DGTcorpora_tokenized/en_gl/wikimatrix/en.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/wikimatrix/gl.txt
transforms: [bpe, filtertoolong]
weight: 70 #25000000/450000 = 55 ; 760/55 = 14 ; 14 * 5 = 70
cluvi:
path_src: ../DGTcorpora_tokenized/en_gl/cluvi/en.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/cluvi/gl.txt
transforms: [bpe, filtertoolong]
weight: 70 #25000000/295000 = 84 ; 760/84 = 9 ; 9 * 10 = 90
#wikimedia:
# path_src: ../DGTcorpora_tokenized/en_gl/wikimedia/en.txt
#path_tgt: ../DGTcorpora_tokenized/en_gl/wikimedia/gl.txt
#transforms: [bpe, filtertoolong]
#weight: 4
# xlent:
#path_src: ../DGTcorpora_tokenized/en_gl/xlent/en.txt
#path_tgt: ../DGTcorpora_tokenized/en_gl/xlent/gl.txt
#transforms: [bpe, filtertoolong]
#weight: 50 #25000000/1600000=15; 760/15=50
#linux:
#path_src: ../DGTcorpora_tokenized/en_gl/linux/en.txt
#path_tgt: ../DGTcorpora_tokenized/en_gl/linux/gl.txt
#transforms: [bpe, filtertoolong]
#weight: 20 #25000000/150000=166; 760/166=5 * 5 = 20
valid:
path_src: ../DGTcorpora_tokenized/en_gl/partitions/all-en_valid.txt
path_tgt: ../DGTcorpora_tokenized/en_gl/partitions/all-gl_valid.txt
transforms: [bpe, filtertoolong]
### Transform related opts:
#### Subword
src_subword_model: ./bpe/en.code
tgt_subword_model: ./bpe/gl.code
src_subword_vocab: ./run/bpe.vocab.src
tgt_subword_vocab: ./run/bpe.vocab.tgt
#src_subword_model: ../sentencepiece/en-gl/en.sp.model
#tgt_subword_model: ../sentencepiece/en-gl/gl.sp.model
src_subword_type: bpe
tgt_subord_type: bpe
src_subword_nbest: 1
src_subword_alpha: 0.0
tgt_subword_nbest: 1
tgt_subword_alpha: 0.0
#### Filter
src_seq_length: 150
tgt_seq_length: 150
# silently ignore empty lines in the data
skip_empty_level: silent
##embeddings
src_embeddings: ../embeddings/en.emb.txt
tgt_embeddings: ../embeddings/gl.emb.txt
## supported types: GloVe, word2vec
embeddings_type: "word2vec"
# word_vec_size need to match with the pretrained embeddings dimensions
word_vec_size: 300
# General opts
save_model: run/model
keep_checkpoint: 50
save_checkpoint_steps: 10000
average_decay: 0.0005
seed: 1234
report_every: 1000
train_steps: 200000
valid_steps: 10000
# Batching
queue_size: 10000
bucket_size: 32768
world_size: 1
gpu_ranks: [0]
batch_type: "tokens"
batch_size: 8192
#batch_size: 4096
valid_batch_size: 64
batch_size_multiple: 1
max_generator_batches: 2
accum_count: [4]
accum_steps: [0]
# Optimization
model_dtype: "fp16"
optim: "adam"
learning_rate: 2
warmup_steps: 8000
decay_method: "noam"
adam_beta2: 0.998
max_grad_norm: 0
label_smoothing: 0.1
param_init: 0
param_init_glorot: true
normalization: "tokens"
# Model
encoder_type: transformer
decoder_type: transformer
position_encoding: true
enc_layers: 6
dec_layers: 6
heads: 8
rnn_size: 512
word_vec_size: 512
transformer_ff: 2048
dropout_steps: [0]
dropout: [0.1]
attention_dropout: [0.1]
share_decoder_embeddings: true
share_embeddings: false
|