Fx

snakers4 · Nov 4, 2018 · 2452643 · 2452643
1 parent 3ba42b0
commit 2452643
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/train_encoder_decoder_predict.py b/src/train_encoder_decoder_predict.py
@@ -112,6 +112,7 @@ def main():
     global USE_CUDA,DEVICE
     global UNK_TOKEN,PAD_TOKEN,SOS_TOKEN,EOS_TOKEN,TRG_NAMES,LOWER,PAD_INDEX,NAMES,MIN_FREQ
     global args,best_met,valid_minib_counter
+    global pieces2text
 
     label_writers = []