forked from MaximumEntropy/Seq2Seq-PyTorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
data_configs.py
33 lines (31 loc) · 842 Bytes
/
data_configs.py
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
training_data_hparams = {
'shuffle': True,
'num_epochs': 1,
'batch_size': 80,
'allow_smaller_final_batch': False,
'source_dataset': {
"files": ['data/iwslt14/train.de'],
'vocab_file': 'data/iwslt14/vocab.de',
'max_seq_length': 50
},
'target_dataset': {
'files': ['data/iwslt14/train.en'],
'vocab_file': 'data/iwslt14/vocab.en',
'max_seq_length': 50
}
}
test_data_hparams = {
'shuffle': False,
'num_epochs': 1,
'batch_size': 80,
'allow_smaller_final_batch': False,
'source_dataset': {
"files": ['data/iwslt14/test.de'],
'vocab_file': 'data/iwslt14/vocab.de'
},
'target_dataset': {
'files': ['data/iwslt14/test.en'],
'vocab_file': 'data/iwslt14/vocab.en'
}
}
valid_data_hparams = test_data_hparams