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

changed batch size to 50 and wts to ans through obj atr in constants.crunchy

parent 834d6a1f
No related branches found
No related tags found
No related merge requests found
......@@ -5,7 +5,7 @@ def mkdir_if_not_exists(dir_name):
if not os.path.exists(dir_name):
os.mkdir(dir_name)
experiment_name = 'obj_atr_through_mil_det_scores'
experiment_name = 'trial_new_rel_feat'
##########################################################################
# Machine Specific Paths #
......@@ -239,15 +239,15 @@ answer_model = os.path.join(
# Answer classifier training params
answer_train_from_scratch = True
answer_batch_size = 25
answer_batch_size = 50
answer_num_epochs = 20
answer_queue_size = 500
answer_regularization_coeff = 1e-5
answer_lr = 1e-3
answer_obj_atr_loss_wt = 1.0
answer_ans_loss_wt = 0.1
answer_mil_loss_wt = 0.1
answer_obj_atr_loss_wt = 0.1
answer_ans_loss_wt = 1.0
answer_mil_loss_wt = 0
# Answer classifier additional joint training params
num_regions_with_labels = 100
......
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