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):
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)
......
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