Sfoglia il codice sorgente

[Transformer/PyT] minor bugfix

Jan Baczek 2 anni fa
parent
commit
7b89aeda73
1 ha cambiato i file con 1 aggiunte e 0 eliminazioni
  1. 1 0
      PyTorch/Translation/Transformer/inference.py

+ 1 - 0
PyTorch/Translation/Transformer/inference.py

@@ -186,6 +186,7 @@ def main(args):
         translator.cuda()
 
     # Load BPE codes file
+    bpe = None
     if args.bpe_codes:
         codes = open(args.bpe_codes, 'r')
         bpe = BPE(codes)