Skip to content

Commit 59ad3db

Browse files
committed
Fix
1 parent 3015e4b commit 59ad3db

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

data_engine/prepare_data.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,7 @@ def build_dataset(params):
219219
fill=params['FILL'],
220220
max_text_len=params['MAX_OUTPUT_TEXT_LEN'],
221221
max_words=params['OUTPUT_VOCABULARY_SIZE'],
222+
bpe_codes=params.get('BPE_CODES_PATH', None),
222223
min_occ=params['MIN_OCCURRENCES_OUTPUT_VOCAB'])
223224

224225
ds.setInput(None, 'val', type='ghost',

0 commit comments

Comments
 (0)