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

select best model passes obj_atr_loss_wt to graph

parent 5093204d
No related branches found
No related tags found
No related merge requests found
......@@ -245,6 +245,7 @@ def eval_model(model_to_eval, results_json):
0,
0,
0,
constants.answer_obj_atr_loss_wt,
resnet_feat_dim=constants.resnet_feat_dim,
training=False)
......
......@@ -159,13 +159,13 @@ num_test_questions = 0
# Answer classifier training params
answer_batch_size = 50
answer_num_epochs = 4
answer_num_epochs = 6
answer_offset = 0
answer_obj_atr_loss_wt = 0.1
answer_regularization_coeff = 1e-5
answer_queue_size = 500
answer_embedding_dim = 600
answer_lr = 1e-3
answer_lr = 1e-4
answer_log_every_n_iter = 500
answer_output_dir = os.path.join(
global_experiment_dir,
......@@ -185,12 +185,12 @@ answer_model = os.path.join(
num_regions_with_labels = 100
# Answer fine tune params
answer_fine_tune_from_iter = 17000
answer_fine_tune_from_iter = 19500
answer_fine_tune_from = answer_model + '-' + str(answer_fine_tune_from_iter)
# Answer eval params
answer_eval_on = 'val'
answer_model_to_eval = answer_model + '-13000'
answer_model_to_eval = answer_model + '-49500'
answer_eval_data_json = os.path.join(
answer_output_dir,
......
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