We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b77976c commit 04a7dd2Copy full SHA for 04a7dd2
scripts/train/train_en_de.sh
@@ -14,7 +14,7 @@ python -m torch.distributed.launch --nproc_per_node=${NUM_GPUS} main.py \
14
'encoder': {'num_layers': 3, 'bidirectional': True, 'num_bidirectional': 1, 'context_transform': 512, 'pack_inputs': False}, \
15
'decoder': {'num_layers': 3, 'concat_attention': True,\
16
'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}" \
+ --data-config "{'moses_pretok': True, 'tokenization':'bpe', 'tokenization_config':{'num_symbols':32000}, 'shared_vocab':True}" \
18
--b 32 \
19
--max-length 50 \
20
--trainer Seq2SeqTrainer \
0 commit comments