Skip to content
Snippets Groups Projects
Commit a5ee2646 authored by tgupta6's avatar tgupta6
Browse files

select_best_model compatible with train

parent 1a015d00
No related branches found
No related tags found
No related merge requests found
...@@ -237,13 +237,14 @@ def eval_model(model_to_eval, results_json): ...@@ -237,13 +237,14 @@ def eval_model(model_to_eval, results_json):
constants.answer_batch_size, constants.answer_batch_size,
constants.image_size, constants.image_size,
constants.num_negative_answers, constants.num_negative_answers,
constants.answer_embedding_dim,
constants.answer_regularization_coeff, constants.answer_regularization_coeff,
constants.answer_batch_size*constants.num_region_proposals, constants.answer_batch_size*constants.num_region_proposals,
0, 0,
0, 0,
0, 0,
constants.answer_obj_atr_loss_wt, constants.answer_obj_atr_loss_wt,
constants.answer_ans_loss_wt,
constants.answer_mil_loss_wt,
resnet_feat_dim=constants.resnet_feat_dim, resnet_feat_dim=constants.resnet_feat_dim,
training=False) training=False)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment