From a5ee2646fbd305146e77f0af629cea1e89c64352 Mon Sep 17 00:00:00 2001
From: tgupta6 <tgupta6@illinois.edu>
Date: Fri, 14 Oct 2016 10:34:45 -0500
Subject: [PATCH] select_best_model compatible with train
---
answer_classifier_cached_features/select_best_model.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/answer_classifier_cached_features/select_best_model.py b/answer_classifier_cached_features/select_best_model.py
index 385c21f..038b5ab 100644
--- a/answer_classifier_cached_features/select_best_model.py
+++ b/answer_classifier_cached_features/select_best_model.py
@@ -237,13 +237,14 @@ def eval_model(model_to_eval, results_json):
constants.answer_batch_size,
constants.image_size,
constants.num_negative_answers,
- constants.answer_embedding_dim,
constants.answer_regularization_coeff,
constants.answer_batch_size*constants.num_region_proposals,
0,
0,
0,
constants.answer_obj_atr_loss_wt,
+ constants.answer_ans_loss_wt,
+ constants.answer_mil_loss_wt,
resnet_feat_dim=constants.resnet_feat_dim,
training=False)
--
GitLab