From 5712f11932042fb36d2e97fe19bf6fbf7056396f Mon Sep 17 00:00:00 2001
From: tgupta6 <tgupta6@illinois.edu>
Date: Fri, 14 Oct 2016 22:40:26 -0500
Subject: [PATCH] answer select best fix batch creator

---
 answer_classifier_cached_features/select_best_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/answer_classifier_cached_features/select_best_model.py b/answer_classifier_cached_features/select_best_model.py
index 9d0d9de..893fc89 100644
--- a/answer_classifier_cached_features/select_best_model.py
+++ b/answer_classifier_cached_features/select_best_model.py
@@ -45,6 +45,8 @@ def create_batch_generator():
         constants.vqa_train_held_out_qids,
         constants.vocab_json,
         constants.vqa_answer_vocab_json,
+        constants.object_labels_json,
+        constants.attribute_labels_json,
         constants.image_size,
         constants.num_region_proposals,
         constants.num_negative_answers,
-- 
GitLab