Error fix #4

carpedm20 · Feb 12, 2016 · d66e2f1 · d66e2f1
1 parent 58f9dee
commit d66e2f1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/batch_loader.py b/batch_loader.py
@@ -153,11 +153,11 @@ def text_to_tensor(self, input_files, vocab_fname, tensor_fname, char_fname, max
               chars.append(char2idx[char])
             chars.append(char2idx['}'])
 
-            for idx in xrange(min(len(chars), max_word_length)):
-              output_char[word_num][idx] = chars[idx]
-
             if len(chars) == max_word_length:
               chars[-1] = char2idx['}']
+
+            for idx in xrange(min(len(chars), max_word_length)):
+              output_char[word_num][idx] = chars[idx]
             word_num += 1
 
         output_tensors.append(output_tensor)