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