Skip to content

Commit 5ac73b4

Browse files
committed
seq2seq data_loader bug fix
1 parent 66004d8 commit 5ac73b4

File tree

2 files changed

+228
-192
lines changed

2 files changed

+228
-192
lines changed

src/11_seq2seq/modules/data_loader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@ def __init__(
3535
batch_first=True,
3636
include_lengths=True,
3737
fix_length=fix_length,
38-
init_token=None,
39-
eos_token=None,
38+
init_token='<BOS>',
39+
eos_token='<EOS>',
4040
)
4141

4242
if train_fn is not None and valid_fn is not None and exts is not None:

0 commit comments

Comments
 (0)