resnet stackGI

KernLC · Jan 31, 2017 · c6f241e · c6f241e
1 parent ed2dde6
commit c6f241e
Show file tree

Hide file tree

Showing 4 changed files with 114 additions and 361 deletions.
diff --git a/data_loader.py b/data_loader.py
@@ -39,7 +39,8 @@
     print(" * %d x %d captions found " % (len(captions_dict), len(lines)))
 
     ## build vocab
-    _ = tl.nlp.create_vocab(processed_capts, word_counts_output_file=VOC_FIR, min_word_count=1)
+    if not os.path.isfile('vocab.txt'):
+        _ = tl.nlp.create_vocab(processed_capts, word_counts_output_file=VOC_FIR, min_word_count=1)
     vocab = tl.nlp.Vocabulary(VOC_FIR, start_word="<S>", end_word="</S>", unk_word="<UNK>")
 
     ## store all captions ids in list