Skip to content

Commit 7b89aed

Browse files
jbaczeknv-kkudrynski
authored andcommitted
[Transformer/PyT] minor bugfix
1 parent d56fe70 commit 7b89aed

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

PyTorch/Translation/Transformer/inference.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -186,6 +186,7 @@ def main(args):
186186
translator.cuda()
187187

188188
# Load BPE codes file
189+
bpe = None
189190
if args.bpe_codes:
190191
codes = open(args.bpe_codes, 'r')
191192
bpe = BPE(codes)

0 commit comments

Comments
 (0)