Skip to content

Commit 04a7dd2

Browse files
authored
Update train_en_de.sh
1 parent b77976c commit 04a7dd2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/train/train_en_de.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ python -m torch.distributed.launch --nproc_per_node=${NUM_GPUS} main.py \
1414
'encoder': {'num_layers': 3, 'bidirectional': True, 'num_bidirectional': 1, 'context_transform': 512, 'pack_inputs': False}, \
1515
'decoder': {'num_layers': 3, 'concat_attention': True,\
1616
'attention': {'mode': 'dot_prod', 'dropout': 0, 'output_transform': True, 'output_nonlinearity': 'relu'}}}" \
17-
--data-config "{'moses_pretok': True, 'tokenization':'bpe', 'num_symbols':32000, 'shared_vocab':True}" \
17+
--data-config "{'moses_pretok': True, 'tokenization':'bpe', 'tokenization_config':{'num_symbols':32000}, 'shared_vocab':True}" \
1818
--b 32 \
1919
--max-length 50 \
2020
--trainer Seq2SeqTrainer \

0 commit comments

Comments
 (0)